Win11下,在Pycharm中安装Cuda和cudnn遇到的一些问题和解决办法记录

        前阵子新买的电脑到货了,RTX3060的显卡,想着终于可以用GPU训练网络了,试着装了一下,这一装就是两天啊。。。

        我没在电脑上装Anaconda,只装了Python和Pycharm,刚开始用pip装了一些像numpy之类的库了。在站里找教程装Cuda,可是装好之后在Pycharm里运行代码

import torch

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

之后,还是显示False。明明要装的都已经装好了,为什么还是这样呢。

        之后又找教程看,感觉自己环境变量可能没配全,就又照着配了一下,还是用不了。后来偶然看到一篇文章http://t.csdn.cn/9NzQw,说是有可能Pytorch没配好,然后发现果然是这样。当时我的pip list里只有torch和torchvision,

就使用

pip uninstall torch

pip uninstall torchvision

卸载了这两个库,

Win11下,在Pycharm中安装Cuda和cudnn遇到的一些问题和解决办法记录_第1张图片

 Win11下,在Pycharm中安装Cuda和cudnn遇到的一些问题和解决办法记录_第2张图片

卸载成功。

        之后照着http://t.csdn.cn/CdJJm 这篇教程去Pytorch官网找了适合自己的pip安装语句,在cmd里进行安装。因为我已经在pip里把搜索源替换为清华大学的软件镜像站了,所以就没想太多,直接运行了。

刚下载一点点就出现了问题 ,换成豆瓣的镜像网站再试

这次快下完了出问题了,真让人头大。在站里找到这篇文章 ,简单修改了一下命令,再次尝试安装。

不知道是学校校园网的问题还是什么原因,连上手机热点又试了一下。想着这次如果再不行,就直接去pip搜索到的这个网站里去直接下载对应的whl包,然后本地安装了。

Win11下,在Pycharm中安装Cuda和cudnn遇到的一些问题和解决办法记录_第3张图片

可算是装好了。激动的心,颤抖的手,打开了我的pip list: 

ohhhhhhhhh~取得阶段性胜利, 然后打开Pycharm,试着跑下程序。Win11下,在Pycharm中安装Cuda和cudnn遇到的一些问题和解决办法记录_第4张图片

终于是using cuda:0 device了,不过出现了OSError: [WinError 1455] 页面文件太小,无法完成操作。 这样的问题。原来是Pycharm装在D盘,没有分配虚拟内存,按照站内文章操作之后,终于成功用GPU跑起了代码,算是配好了环境。

总结

  1. 安装Python和Pycharm。
  2. 如果装过Pytorch的话,一定要确定自己装的是不是GPU的版本,如果不是,要想办法卸载。
  3. 查看自己电脑的CUDA版本,以此为依据安装对应版本的CUDA和cudnn,把cudnn的文件夹直接粘贴到CUDA的安装目录中。
  4. 在Pytorch官网查找适合自己的安装命令进行安装。
  5. 如果安装中途出现问题,可以尝试换源或者使用timeout的方法再次尝试,实在不行可以在网站里直接下载,然后手动本地安装。
  6. 如果还有问题,要看看自己的环境变量全不全。

我的电脑环境

Win11家庭中文版21H2

Pycharm: Community 2022.2.1

Python:3.10.6

GPU: RTX3060

CUDA: 11.6.112

 cudnn: 8.5.0.96

你可能感兴趣的:(pycharm,python,pytorch)