4070配置pytorch-GPU过程记录

为新买的电脑配置深度学习环境,记录几个踩坑点

我的电脑信息:4070显卡,最高支持CUDA12.0,已安装pycharm,anaconda并且已经创建环境python=3.10

在安装CUDA之前需要先确定pytorch支持的CUDA,截止到2023.3.3号pytorch最高版本先行版支持11.8(稳定版是11.7,但搜索资料得知pytorch是从11.8版本才开始支持40系显卡);

所以要安装的CUDA版本为11.8,并在英伟达官网下载安装对应cudnn版本为8.8。CUDA与cudnn的安装过程参考大佬链接:CUDA与cudnn安装

安装完成之后开始下载pytorch-GPU:

到pytorch官网中找到先行版11.8:

此时请注意: 若你已经将conda的下载源更换成清华的镜像源,请不要使用conda指令下载!!使用conda安装会自动下载cpu版本的pytorch,因为清华源的pytorch里没有支持CUDA11.8的。一定要注意,这里坑了我一天。

此时安装方法有三个:

1.将清华源移除后在返回官网使用conda下载:

移除指令:

conda config --remove-key channels

2.使用pip下载:

pip下载时如果出现反复下载pytorch的情况,原因是python的版本不匹配,pytorch-CUDA11.8支持的python版本为>=3.10。(此处附上pip安装包的路径:C:\Users\xhb\AppData\Local\pip,因为它会占用C盘空间)

3.先下载安装包然后去cmd中安装:

下载地址就是pip安装指令后面的网址,这里是

https://download.pytorch.org/whl/nightly/cu118;

 下载内容是pip指令的前面部分,这里是

torch torchvision torchaudio

下载完成之后在下载目录中进入cmd,并激活环境(要用cmd,不要用anaconda prompt)

然后依次安装上述三个下载好的文件:

pip install torch-2.1.0.dev20230302+cu118-cp310-cp310-win_amd64.whl
pip install torchaudio-2.0.0.dev20230302+cu118-cp310-cp310-win_amd64.whl
pip install torchvision-0.15.0.dev20230302+cu118-cp310-cp310-win_amd64.whl

总结:三个方法中我使用了第三种方法安装成功,第一种方法下载太慢最后直接停止导致安装失败;第二种方法我在将python的版本改成3.10后还是会重复下载torch,没搞懂为啥;第三种方法过程略显麻烦,但是好用。

最后验证是否GPU可用:

python
import torch
torch.cuda.is_available()

结果为True则安装成功!

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年6月9日
下一篇 2023年6月10日

相关推荐