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

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
安装完了CUDA11.1 并且配置好环境变量后,测试

import torch
print(torch.__version__)
print(torch.cuda.is_available())

最后输出的结果是true!
可以愉快的进行网络训练了

(坐等pytorch出CUDA11.2的版本)

你可能感兴趣的:(小猫深度学习,深度学习,gpu,pycharm,cuda)