cuda10.0环境安装Pytorch
前置环境需要安装cuda10.0和对应的cudnn不在赘述,如需查看请参考这两篇博客(面向小白的,十分详细):手把手教你win10安装Paddlepaddle-GPU(一)手把手教你win10安装Paddlepaddle-GPU(二)尴尬的是,官网好像比较支持CUDA9.2和CUDA10.1,没有CUDA10.0的一席之地…从官网中找到稳定版下载的网址https://download.pytorc
前置环境需要安装cuda10.0和对应的cudnn
不在赘述,如需查看请参考这两篇博客(面向小白的,十分详细):
手把手教你win10安装Paddlepaddle-GPU(一)
手把手教你win10安装Paddlepaddle-GPU(二)
尴尬的是,官网好像比较支持CUDA9.2
和CUDA10.1
,没有CUDA10.0
的一席之地…
从官网中找到稳定版下载的网址
https://download.pytorch.org/whl/torch_stable.html
搜索cu100,可以发现,Pytorch对windows的cuda10.0可真是太不友好了
最高版本也只是支持到Pytorch1.1.0
,真的尴尬
通过查看Pytorch
和torchvision
的对应关系:
https://blog.csdn.net/qq_40263477/article/details/106577790
我们下载torchvision 0.3.0
就可以了
这两个下载链接分别是:
pytorch:https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
torchvision:https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
扔到某雷里边下载,速度还行
如果链接变了,大家从那个稳定版的下载链接中找一下就行
最后验证一下:
import torch
torch.cuda.is_available() # True 就完事儿了
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)