Pytorch 2.3.x最新版本安装教程（适合有基础的看）

这几天换新电脑，重新安装Pytorch，遇到的一些问题，记录一下。具体安装步骤大家可看https://blog.csdn.net/MCYZSF/article/details/116525159，前面步骤一样，不做过多赘述，大家按照原文的一步步来就好，主要是最后官方下载torch问题

Pytorch官方安装

主要问题是，在2.3.1的版本官方链接没有给出版本信息，下载的东西在一个库里，通过换源等方式下载安装下来的，以及使用pip在清华源下载的，都是得到cpu版本，与cuda无法使用。

 库中内容

torch

torchvison

torchaudio

解决方法

1、继续使用官方命令，但需要开启梯子，不然网络不稳定容易出错

2、使用命令pip install torch==“torch版本号”+“cuda版本号”  torchvision==“vision版本号”+“cuda版本号” torchaudio==“版本号”

具体看上图库中的内容，比如我的cuda是12.5，选择个向下兼容的，选取12.1，表示为cu121，torch要2.3.0的版本，python的版本是3.9.所以cp为39，根据这种规则撰写自己版本的下载命令，然后加入清华源

pip install torch==2.3.1+cu121 torchvision==0.18.1+cu121 torchaudio==2.3.1 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

 不想麻烦的直接使用官方命令，加个梯子，避免出错最好的方案，本文请结合上文中的链接搭配，主要描述torch 2.3.x在最后安装过程中的错误，检测是否成功的代码如下：

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())  #输出为True，则安装无误

成功安装output：