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

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

这不是写了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/

  • 29
    点赞
  • 114
    收藏
    觉得还不错? 一键收藏
  • 18
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值