前言:电脑挺久没用,可能由于版本更新问题,import torch的过程异常缓慢。寻思着重新配个环境,跟着网上的教程走。大致都是两种流程,一种就是从pytorch官网下载,一种是用镜像源。前者基本要等个两三小时,后者也不知道为什么,下载速度也慢的要死,还没有最新版本的torch。捣鼓了半天,发现还是直接下whl省事。
要求:anaconda、科学上网工具、pycharm
anaconda和pycharm的安装网上有很多教程我就懒得细讲了(主要这次我自己也没重新搞这两个,没过程)。这里从anaconda和pycharm已经配置好的地方开始讲起。
跟着其他教程发现下载pytorch速度慢,或者镜像源无法下载,想要直接下whl文件的的朋友直接跳到步骤四即可。
一、conda创建虚拟环境
在自己的电脑中找到conda命令行
打开它,输入
conda create -n pytorchNew python=3.9
注意,pytorchNew为你想要创建的环境的名字,自己可以改,在这篇文章的时间点,python使用3.9,3.10,3.11,3.12都可,不能小于3.9,具体要求要去pytorch官网查看,具体会在第三步中说明
遇到上图问你yes或no,输入y回车。
然后激活该环境,使用
conda activate 环境名字
二、检查自己的cuda版本(无显卡的可跳过这步)
在上述命令行中输入
nvidia-smi
回车后查看自己的cuda版本
三、尝试pytorch官网方法安装
这一步仅作为尝试,若成功的话说明你的网络很好,若发现速度极其缓慢请及时使用ctrl+c退出。
打开pytorch官网下载链接:Start Locally | PyTorch
往下滑找到这里:
可以看到图片第一行告诉了我们python版本的要求,这就是我说的第一步中创建环境需要考虑的内容。
1、若你的电脑无独立显卡,选择CPU即可
2.1、若要安装GPU版pytorch,选择第二步中你查看到的CUDA版本对应的版本,如果没有,选择不高于你的CUDA版本的最近的版本,这玩意向下兼容的(比如我的CUDA版本为12.8,那我可以安装不高于12.8的所有版本,一般都是安装最新的,所以选择12.6版本的)(以上所说的版本都是pytorch的CUDA版本,不是pytorch的版本)
2.2、有人这时候又有问题了:那我的的CUDA版本低于11.8怎么办?可以考虑更新显卡驱动,我的卡是N卡,可以直接使用GeForce Experience更新,到官网下载应用打开它就能看到更新驱动的选项:GeForce Experience 可自動更新驅動並優化遊戲設置 | NVIDIA,其他的卡更新驱动的话自己网上随便搜一下教程就有。
选好选项后复制最后一行的命令:
输入到自己的conda命令行里面再enter即可(红框标注意思是注意要在对应的环境下):
如果也遇到下图这种要下两个多小时的情况:
完全正常,ctrl+c直接打断
四、科学上网直接到官网下载whl文件
挂上vpn,回到我们之前说的pytorch下载网站:Start Locally | PyTorch
找到这里,复制图中红笔标注所示的链接:
(不管你的CUDA是什么版本,或者是用cpu版的pytorch,都可用直接用这个链接,打开的网站是一样的。)
复制到浏览器打开,进入到如下网站:
我们常说的pytorch,其实就是torch,torchvision,torchaudio这三个包(从官网提供的下载命令也能看出来),接下来就是使用这个网站下载这三个包的whl文件的流程
1、先下载whl文件。
在上面的网站中ctrl+f,搜索torch,找到torch后点进去:
你会进入到以下界面:
还是ctrl+f进行搜索,这里根据你需要的torch的CUDA版本,输入相应的数字(去掉小数点),比如我要下的是CUDA12.6的版本,所以搜索126:
cp后面的数字代表python版本,我的环境python=3.9,选择cp39的,最后面的代表电脑类型,win下win的,Linux下Linux的。综上,我需要下载的就是torch-2.6.0+cu126-cp39-cp39-win_amd64.whl。
点击一下就自动下载了,但是可以看到下载速度也很慢:
这也是为什么需要科学上网的原因,我们打开vpn后的速度:
要安装cpu版本的,可以ctrl+f然后搜索cpu-cp39(这个39按照你的python版本替换),下载最新的torch即可。
torchaudio、torchvision的下载也差不多,回到我们之前找到torch的地方:
点击torchaudio,然后搜CUDA号(比如126),cpu版本参照搜cpu-cp39,再下载对应的版本即可。torchvison也一样。
2、安装whl文件
先把vpn关掉,防止安装依赖包时出现错误。
现在,我们已经下好了对应的whl文件,在文件目录中找到它们:
复制文件路径,在conda命令行中使用以下命令完成安装:
pip3 install 文件路径
示例(注意下载后文件的位置和我这的可能不太一样,自行修改文件路径):
(pytorchnew) C:\Users\86175>pip install D:\Download\torch-2.6.0+cu126-cp39-cp39-win_amd64.whl
(pytorchnew) C:\Users\86175>pip install D:\Download\torchaudio-2.6.0+cu126-cp39-cp39-win_amd64.whl
(pytorchnew) C:\Users\86175>pip install D:\Download\torchvision-0.21.0+cu126-cp39-cp39-win_amd64.whl
3、验证安装
conda命令行中输入python进入到python编译环境,再
import torch
print(torch.cuda.is_available())
结果为True,即安装成功。
五、在pycharm中导入环境
打开pycharm,在右下角你会看到当前环境:
点击后,再点击添加解释器,打开如下弹窗;
左侧选择conda环境,再选择现有环境,下拉列表选中我们创建的环境,确定后即可:
然后我们就能再环境列表中找到它了:
选中切换即可。