GPU版的Pytorch、torch、torchvision、torchaudio安装（用pip，无conda）_pip install torch==2.1.0 torchvision==0.16.0 torch

1查看python对应的pytorch版本

pytorch的github官方文档：

https://github.com/pytorch/vision#installation

只要根据自己的python版本，选择torch就行，torchvision在之后用pytorch的官网的命令可以直接确定（第二步会讲）

近期pytorch版本：

往期pytorch版本：

2查看CUDA版本

输入命令nvcc -V

nvcc -V

如图，我的CUDA版本为11.1

注：命令nvidia-smi查看的是自己电脑显卡驱动下支持的CUDA最高版本，不是当前电脑上有的CUDA版本！如下图，我的电脑能支持的最高版本为11.6，那么我安装的CUDA版本不能超过这个数，而我安装的CUDA版本为11.1，没有超过。

nvidia-smi

3 查看CUDA对应的pytorch版本安装 

还记得我们在第一步查看了python对应的pytorch版本吗，也就是说，要安装的pytorch必须同时满足python版本和CUDA版本，两者取交集哦

先看链接，还不急着安装

pytorch最新版本：https://pytorch.org/get-started/locally/

pytorch往期版本：https://pytorch.org/get-started/previous-versions/

4使用pip本地安装

注：直接把官网的命令粘贴到命令行可能安装不成功，我的情况是会报WARNING，并显示找不到相应版本，网上看有的网友们似乎anaconda虚拟环境能成功。因为此处不用conda，因此通过wheel进行安装。

如图，根据CUDA版本找到相应命令

采用wheel方式安装，需要每个库单独下载对应的wheel，分别安装，黄框内标注了此次要下载的三个库:torch torchvision trochaudio

接下来以torch为例进行安装

​
（1）下载wheel

下载地址：

https://download.pytorch.org/whl/cu111/

推荐迅雷下载，一般浏览器下载会比较慢

（2）pip install安装torch

pip install 【刚刚下载的torch的wheel的地址】

重复上述两小步，安装剩下的torchvision、torchaudio

5 检测pytorch是否安装成功

cmd输入pip install

pip install

再分别输入以下命令

python
import torch
torch.cuda.is_available()

如上图，出现True就算安装成功啦！