我的cuda9.2,实测可以安装9.0版本
查看自己cuda版本的方法:
控制面板-硬件和生音-NVIDIA控制面板-帮助-系统信息组件
9.2.217
参考:https://blog.csdn.net/weixin_42158966/article/details/88543668?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task
具体安装过程
创建虚拟环境
conda create -n pytorch python=3.6
激活
activate pytorch
添加清华镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes
虽然我的版本是9.2,官网推荐的也是下面的程序写9.2,但清华镜像源没有这个版本(2020.3.30)
所以我装了9.0
conda install pytorch torchvision cudatoolkit=9.0
亲测可用
参考:https://blog.csdn.net/qq_31736627/article/details/99415460?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task