WIN10+3090+CUDA11.2+cudnn8.2+python3.7+pytorch1.9配置unet（未完记录）

最新推荐文章于 2024-08-10 17:44:40 发布

bozki1995

最新推荐文章于 2024-08-10 17:44:40 发布

阅读量2.8k

点赞数 2

分类专栏：小猫深度学习文章标签：深度学习 gpu pycharm cuda

本文链接：https://blog.csdn.net/bozki1995/article/details/118570158

版权

本文记录了使用Win10系统，配备NVIDIA 3090显卡，尝试配置CUDA11.2和cudnn8.2以运行基于PyTorch1.9的unet模型的过程。在安装过程中遇到PyTorch官网缺少CUDA11.2对应版本的问题，通过降级CUDA到11.1并解决兼容性问题，最终成功使torch.cuda.is_available()返回true，能够利用GPU进行深度学习训练。

摘要由CSDN通过智能技术生成

WIN10+3090+CUDA11.2+cudnn8.2+python3.7+pytorch1.9配置unet（未完记录）

显卡英伟达3090，看了英伟达控制面板里面对应的cuda版本是11.2
提前装好了VS2019，然后再配好了CUDA11.2，以及对应的cudnn8.2
根据我想做的项目，我装了python3.7，这些安装过程就略去了
在跑代码的时候发现了没有调用GPU，发现了torch.cuda.is_available()=false
应该是下载的pytorch和cuda版本不对应
然后我去pytorch官网下载，发现居然没有对应cuda11.2的版本！！！无语
现在摆在我面前的有两个办法，一个是降cuda版本，一个是下pytorch cuda11.1的版本试试。
于是我先试了第二种
pip3 install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

等我试好了再更新！！！

UU们我回来了！！！
事实证明，cuda11.2并不能和这个1.9的pytorch兼容，气死，说好的cuda向下兼容能力强呢
为了不浪费3090的能力，我又安装了CUDA11.1，下载了适合CUDA11.1的CUDNN，让两个版本的CUDA同时存在
参考：https://blog.csdn.net/L1778586311/article/details/112398545