cuda版pytorch兼容问题

在下载GPU版pytorch前必须先下载cuda和cudnn。建议cuda11.3+cudnn8.2.1.3+pytorch1.10(目前官网首页),这样我运行下来没有任何问题。

下载cuda

先看电脑显卡支持的cuda版本,注意:这里查到的是最高版本,你可以选择下载低版本。

win+R 输入cmd  打开窗口后输入:nvidia-smi 

我的最高版本是11.6 ,那么11.6前的都可以下载

我下载的11.3 链接:​​​​​​CUDA Toolkit 11.3 Downloads | NVIDIA Developer

步骤见下图,下载好直接安装。

安装好寻找合适版本的cudnn,直接选择cuda11.3对应的版本,但是你进去发现没有,所以建议11.x版本

cuDNN Download | NVIDIA Developer

 下载cuDNN需要注册账号,注册很简单,就不多说了。

找之前的版本,红线这些11.3都是可以用的。但是最好建议和我的一样。这样为了避免很多麻烦。

 下载好对应的cudnn就看下边这个教程,把该复制的复制到CUDA安装包里就好,主要,CUDA大家基本都是在C盘programFIie下面。配置:

复制 cuDNN bin 目录下的文件到 CUDA 的 bin 目录下(.dll)
复制 cuDNN include 目录下的文件到 CUDA 的 include 目录下(.h)
复制 cuDNN lib/x64 目录下的文件到 CUDA 的 lib/x64 目录下(.lib)
添加环境变量,把 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64 加到path 中  原文链接:https://blog.csdn.net/weixin_38673554/article/details/90513732

 下面说一下pytorch,建议直接进官网,直接首页

 将下面的话复制到Anaconda Prompt中,最好建个单独的分区放pytorch、tensorflow这些,详情看:(23条消息) PyTorch安装最全流程_李坦(BNU远程教育学)的博客-CSDN博客_pytorch安装步骤

这个真的具详细,建议小白直接按照他流程,不过版本可能有些老了,所以还是要看我这个对应的版本。下载时候我是没有出现pytorch太大下不下来的问题,就是很慢,但是你就等呗,中途不能中断服务!!!!!!!不然要重新下。、

最后在你的pycharm里测试一下gpu版的pytorch成功没成功就好啦‘。

# 依次输入
import torch
torch.cuda.is_available()

True代表成功了。

  • 4
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值