5365bet手机版客户端

30分钟内安装好pytorch(不用镜像,需要科学上网)

30分钟内安装好pytorch(不用镜像,需要科学上网)

前言:电脑挺久没用,可能由于版本更新问题,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环境,再选择现有环境,下拉列表选中我们创建的环境,确定后即可:

然后我们就能再环境列表中找到它了:

选中切换即可。

相关推荐