我的安装环境是 win10 + cuda 10.0 + anconda
之前去官网安装,发现太慢了,而且对应版本不对,官网下载的对应cuda选择只有 9.2,10.1,10.2
这里还有个坑是 查看自己电脑的cuda对应版本,我忘记之前下载的cuda版本了,所以想再查看一下
方法1:进入NVIDIA控制面板,选择进入导航栏的帮助中的系统信息,点击组件查看。
这种方式显示我的cuda版本是10.1,这个应该是系统自带的版本
方法2:cmd在命令行输入 nvcc --version
这种方式显示我的cuda版本是10.0
然后去官网 https://pytorch.org/get-started/locally/
得到版本网站 https://download.pytorch.org/whl/torch_stable.html
选择自己适合的版本,下载之后 pip安装
最开始,我选择的是 torch1.2.0 和 torchvision 0.4.0
但是安装完之后发现,会出现 “AttributeError: 'torch._C.Value' object has no attribute xxx” 的问题
查了很久,发现可能是 torch 1.2.0的问题
然后重新下载 安装了 torch 1.1.0 和 torchvision 0.3.0
这下 可以 import torch了
但是 import torchvision 的时候出了问题
经过下面链接处理,终于可以成功导入了