写在前面:这是一个完整的利用镜像源安装torch的方法。本人是小白,所以写的很基础很详细。
我自己安装的时候参考了下列帖子,非常感谢大佬们的分享:
conda安装pytorch出现Solving environment: unsuccessful initial attempt using frozen solve.-优快云博客
【真正解决】Conda安装卡在solving environment这一步!-优快云博客
基本上看了这俩帖子就可以自己安装了,不过我还是想详细的完整的写一下安装过程(毕竟我花了3天才装好,踩了好多坑)。
首先你的电脑要装有anaconda,并建立好一个虚拟环境。我们将在这个虚拟环境里安装torch。
一、创建虚拟环境
首先,打开anaconda prompt,并激活/创建一个虚拟环境。
创建虚拟环境(这里虚拟环境我命名为test,后面python=x.x指的是python版本,这个可以自己定义):
conda create -n test python=x.x
二、换源
好的,最重要的时刻来了。有些人之前可能为了提高下载速度装了很多源,正如开头参考贴中所说,这些杂乱的源可能是torch安装不成功的原因。
所以,首先我们需要恢复默认源:
conda config --remove-key channels
在此之后,我们需要安装新的镜像源(from开头参考贴):
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/main
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/r
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/msys2
conda config --set show_channel_urls yes
然后,有一个非常重要的动作!打开你的C盘,找到C盘下user下的.condarc文件,并用记事本打开它,删掉-default(下图是我自己删过的):
三、确认本机CUDA版本&找到torch官网安装命令
CUDA确认方法:同时按住windows+r键,在弹出的框里输入cmd,回车。在黑色命令窗口输入nvidia-smi,就会看到黑色的命令窗口里出现了一些信息,如下图。可以看到第一行右侧有个CUDA Version:12.2,这就是该电脑的CUDA版本。
(解释:确认CUDA以确定接下来torch安装哪个版本)
torch安装命令:搜索torch,进入torch官网(进不去可能需要科学上网),然后我们下拉,可以看到如下界面,安装命令我贴在下面了(其中,stable是稳定版本;window是系统不多说了;conda是,,,我理解是安装方式的意思,不知道对不对,反正选它就行;python是语言;最后这个Compute platform一般是要选和自己电脑CUDA版本一致的才对,但是12.2目前还没有对应的,因此先装一个12.1的,据说是可以用。):
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
四、安装torch
激活虚拟环境(test为虚拟环境名称):
conda activate test
然后把上述安装命令直接贴上去,不需要删除任何东西,直接回车运行。等电脑自己装,大概十几分钟,应该就能装好。
欢迎交流。