超坑!安装CUDA与pytorch版本匹配的问题

慢悠悠地装了很久,觉得CUDA和pytorch官网标注的当前该pytorch所匹配的CUDA应该是完全一致的,所以一开始就安装 10.1.105版本的CUDA,pytorch为1.4(官方说适配CUDA版本为10.1嘛~):

超坑!安装CUDA与pytorch版本匹配的问题_第1张图片
但是事实证明,不行
最后还是降级安装pytorch,从1.4(对应CUDA10.1)到1.2版本的pytorch(对应CUDA10.0),还是不行,最后:

显卡驱动版本号:431.56(安装之前一定要看自己的显卡驱动版本号,然后对应下载CUDA与cudnn进行安装~)
CUDA版本:10.1.105
cudnn版本:10.1
pytorch版本:1.1.0 (对应CUDA 10.0)

测试成功!!
超坑!安装CUDA与pytorch版本匹配的问题_第2张图片
所以如果安装好了CUDA之后,可以选择pytorch官方说明的适配CUDA版本低一些的来进行安装~

你可能感兴趣的:(学习备忘)