Pytorch血泪安装史好吗(GPU版本+cuda12.1+python3.9.13)

1.安装cuda

首先看下自己电脑是CPU还是GPU,看自己电脑对应的cuda版本  看右下角英伟达标识,点击组件,我的cuda版本是12.3,但最后发现安12.1比较好

2.安装12.1cuda版本对应的cudnn

前两步可以看参考:Pytorch的安装,有点繁琐但是很详细,保姆级教程不信你安装不成功(Cuda+Cudnn+Anaconda+Pytorch)_pytorch安装-CSDN博客

3.anaconda安装以及环境变量配置

①anaconda安装

注意不要安最新的版本,别问我为什么(可能不太好找对应的pytorch版本,太新也容易和很多软件不兼容),尽量安3.9版本,我用的是3.9.13

anaconda版本下载:Index of /

部分版本对应:

②配置环境变量

我放到D盘了,所以必须要设置环境变量

D:\APPS\anaconda3
D:\APPS\anaconda3\Scripts
D:\APPS\anaconda3\Library\bin

最后可以命令行输conda

有结果显示就是成功了

4.比较重要的就是anaconda的镜像源的配置和虚拟环境建立

①镜像源配置,配置完记得再检查下镜像源列表(这里好像环境激活前提前配置也可以,你们可以试一下)

conda config --show channels
# 查看当前镜像源有哪些
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
# 添加镜像源 别问我为什么只给你一个 因为中间真的出现了太多问题(或者直接修改用户盘的.condarc文件)

②虚拟环境建立:

首先要知道自己的python版本,可以先查询一下:

conda create -n 虚拟环境名 python==3.9.13

输入 y

记得建立完最后要激活一下:

conda activate 你的虚拟环境变量名

最后输入虚拟环境列表查询,检查下列表是否有刚才的虚拟环境名字:

conda env list

③pytorch下载!!!记住这里是Gpu版本!!!!!

拿不准可以去官网搜一下适配的版本

激活虚拟环境后,输入

conda install pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=12.1 -c pytorch -c nvidia

最后检查一下是否下载成功

进入anaconda prompt

激活虚拟环境 conda activate 你的虚拟环境名称 

输入python

输入import pytorch

再输入torch.cuda.is_available()

④ 虚拟环境切换盘

最后记得将存放在c盘的虚拟环境转移到D盘anaconda下:Win11转移conda环境——从C盘转移到D盘_anaconda怎样从c盘移动到d盘-CSDN博客

 如图所示:

5.pycharm中更新解释器

①打开pycharm,新建工程,重新加载工程位置。

②点击右下角解释器,asd 

②依次点击

③ 找到你的anaconda,找到python.exe ,选中点击 ok

新建工程或者添加解释器的时候注意看解释器名称,不是base啊,是我们之前新建立的虚拟环境名称,点击切换可以了

复盘:

1.问题比较多在于镜像源出了很多问题,一会清华不行,一会中科大不行,记得选带anaconda的,报错了就直接百度

2.很多攻略对于anaconda虚拟环境中pytorch的下载,没有区分GPU和CPU版本。导致pytorch虽然pycharm中不影响使用,但是却在虚拟环境中显示不出

3.对于版本的匹配,pytorch和python、cuda都是较严格的。最好去官网搜适配的三个版本,然后直接下,不然后续用起来会有很多问题。

  • 12
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值