超坑!安装CUDA与pytorch版本匹配的问题 |
您所在的位置:网站首页 › pytorch版本过低 › 超坑!安装CUDA与pytorch版本匹配的问题 |
慢悠悠地装了很久,觉得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/ |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |