windows10 , python3.6.5 ,rtx3060ti
一:问题及初步尝试解决
最近跑一个需要使用cuda的代码,但是一直有问题。检查到最后发现是最前面有一个if判断cuda是否可用一直返回false
if torch.cuda.is_available()
这说明cuda不可用,于是选择输出torch和cuda的version看一下,即
print(torch.__version__)
print(torch.version.cuda)
发现前一个输出结果是None,后一个输出结果是'1.10.2+cpu',这说明pytorch装的版本是CPU版本的。于是去卸载虚拟环境中的torch,去pytorch官网安装cuda11.7的pytorch,如图:
在cmd中进入虚拟环境后输入以上指令:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117安装后运行程序返回依然是false,我怀疑是cuda和cudnn的问题,于是重新走一遍过程。当然尝试了多种办法都没用后,我最后确定是pytorch安装的问题,是pytorch官网给的指令安装的并非合适版本的cuda11.7的pytorch,于是只好向下适配,去安装了cuda11.3的版本,然后虚拟环境安装了对应的包,最后运行成功,以下是流程。
二.安装cuda和cudnn
进入cuda官网
https://developer.nvidia.com/cuda-toolkit-archive
找到合适版本
这里我安装的11.3.0版本
选择自定义且cuda选择去掉visual studio integration,除了cuda,其他的也去掉勾选(因为我有cuda11.7版本,其他不需要的都不安装)
然后下载cudnn:https://developer.nvidia.com/rdp/cudnn-download
安装好后解压缩,放到对应的文件夹中,即将安装好后的文件每个放入对应文件夹中,找到路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3的文件夹,然后将bin,include,lib中的文件一一放过去复制粘贴即可。
此时cuda和cudnn就安装完毕。
三.安装pytorch
去pytorch官网找到Previous versions of Pytorch
找到cuda11.3的指令去虚拟环境中输入指令:
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
然后安装完成后去验证代码:
问题解决。文章来源:https://www.toymoban.com/news/detail-453457.html
多版本cuda可共存,参考博客:https://blog.csdn.net/m0_37605642/article/details/117932717文章来源地址https://www.toymoban.com/news/detail-453457.html
到了这里,关于解决cuda版本不对应问题的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!