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都是较严格的。最好去官网搜适配的三个版本,然后直接下,不然后续用起来会有很多问题。