慢悠悠地装了很久,觉得CUDA和pytorch官网标注的当前该pytorch所匹配的CUDA应该是完全一致的,所以一开始就安装 10.1.105版本的CUDA,pytorch为1.4(官方说适配CUDA版本为10.1嘛~):
但是事实证明,不行。
最后还是降级安装pytorch,从1.4(对应CUDA10.1)到1.2版本的pytorch(对应CUDA10.0),还是不行,最后:
显卡驱动版本号:431.56(安装之前一定要看自己的显卡驱动版本号,然后对应下载CUDA与cudnn进行安装~)
CUDA版本:10.1.105
pytorch版本:1.1.0 (对应CUDA 10.0)
注意:本身的CUDA版本与我们在虚拟环境中安装的cudatoolkit包是没有太大关系的,一般安装pytorch时需要考虑的cuda版本指的应该是虚拟环境中安装的cudatoolkit包的版本(个人经验激情发言,不负责任噢)。
测试成功!!
所以如果安装好了CUDA之后,可以选择pytorch官方说明的适配CUDA版本低一些的来进行安装~
目前俺正在使用的两种版本是:
①
cudatoolkit:10.1.243
pytorch1.6.0(pytorch1.3.1也可以)
cudnn是7.6.4
②
cudatoolkit10.0.130
pytorch:1.1.0
附:安装tensorflow-gpu版本:
conda install tensorflow=1.13 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/