pytorch的安装（GPU版本和CPU版本都可以）_torch 2.1.2

pytorch的安装（GPU版本和CPU版本都可以）

直接去download.pytorch.org/whl/torch_stable.html该网站下载torch，如下图所示:

其中，前面的cu表示cuda，118表示cuda的版本号为11.8，torch-2.1.0表示torch的版本号为2.1.1，cp38为python版本为3.8，310表示python版本为3.10，win、linux代表不同的系统。

cpu版本

注意安装时候看自己的电脑有没有GPU，如果没有的话安装gpu版本应该会出现问题，要安装cpu版本。但安装cpu版本建议可以直接

pip install torch -i https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

安装会更加快捷。

gpu版本

若不知道自己的cuda版本号，windows系统可以打开命令行输入nvidia-smi，即可查看，如下所示:

python的版本号可以直接python -v进行查看

知道了自己电脑cuda和python的版本号之后，就可以进入上面所写的链接中，进行下载。

注意:上文中讲到的版本号要对应好，不然很容易出现错误！！！

下载完成之后，就可以使用pip instasll 的方式来进行安装whl文件，若安装的环境为虚拟环境的话，记得要激活自己的虚拟环境，然后在虚拟环境下面进行

pip install 文件名

例：pip install torch-1.9.0+cu111-cp39-cp30-win_amd.64.whl

注意(我这个写的默认已经在此文件的安装目录了，如果不在，要么选择cd到这个目录，要么在此文件的前面加上路径名如”pip install e://test/torch-1.9.0+cu111-cp39-cp30-win_amd.64.whl“)

注意

下面写的这个方法，我没有亲自试，建议大家还是直接装自己版本所对应的，实在找不到再采用下面的这种方式。

若出现

whl is not a supported wheel on this platform

则可以pip debug --verbose命令查看平台支持的版本

对比我们的文件名，发现不在这个支持的版本列表里面，很简单，修改whl文件名即可，我修改后的文件名如下：torch-1.12.1+cu116-cp39-abi3-win_amd64.whl，其实对照上图中的任何一个的名称都可以，然后再用pip install torch-1.12.1+cu116-cp39-abi3-win_amd64.whl 安装，成功。

安装成功之后可以

import torch
print(torch.cuda.is_available() )
1
2

若输出为true则代表torch的gpu版本就安装成功了，若为false，表示gpu不可用。

参考链接：

若输出为true则代表torch的gpu版本就安装成功了，若为false，表示gpu不可用。

参考链接：

用whl安装pytorch踩坑实录（whl is not a supported wheel on this platform.）_安装pytorch失败-CSDN博客