前阵子新买的电脑到货了,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
卸载了这两个库,
卸载成功。
之后照着http://t.csdn.cn/CdJJm 这篇教程去Pytorch官网找了适合自己的pip安装语句,在cmd里进行安装。因为我已经在pip里把搜索源替换为清华大学的软件镜像站了,所以就没想太多,直接运行了。
刚下载一点点就出现了问题 ,换成豆瓣的镜像网站再试
这次快下完了出问题了,真让人头大。在站里找到这篇文章 ,简单修改了一下命令,再次尝试安装。
不知道是学校校园网的问题还是什么原因,连上手机热点又试了一下。想着这次如果再不行,就直接去pip搜索到的这个网站里去直接下载对应的whl包,然后本地安装了。
可算是装好了。激动的心,颤抖的手,打开了我的pip list:
ohhhhhhhhh~取得阶段性胜利, 然后打开Pycharm,试着跑下程序。
终于是using cuda:0 device了,不过出现了OSError: [WinError 1455] 页面文件太小,无法完成操作。 这样的问题。原来是Pycharm装在D盘,没有分配虚拟内存,按照站内文章操作之后,终于成功用GPU跑起了代码,算是配好了环境。
Win11家庭中文版21H2
Pycharm: Community 2022.2.1
Python:3.10.6
GPU: RTX3060
CUDA: 11.6.112
cudnn: 8.5.0.96