快速安装GPU版本PyTorch(pip/阿里云/CUDA 12.1)

1. 安装cuda 12.1版本pytorch

pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 -f https://mirrors.aliyun.com/pytorch-wheels/cu121

2. 验证是否安装成功

python
>>> import torch
>>> torch.cuda.is_available()
True

3. 其他版本尝试

pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 -f https://mirrors.aliyun.com/pytorch-wheels/cu121
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 -f https://mirrors.aliyun.com/pytorch-wheels/cu121
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 -f https://mirrors.aliyun.com/pytorch-wheels/cu121

特别注意：这里针对的是cuda 12.1版本，若需其他版需查看pytorch历史版本对应torchvision、torchaudio版本；同时链接“https://mirrors.aliyun.com/pytorch-wheels/cu121”中"cu121"一般表示cuda 12.1，需要注意对应。

4. 查看本机cuda版本

# 方法1
nvidia-smi
# 方法2
nvcc -V

注：一般安装pytorch版本对应cuda版本，不能超过以上命令查阅得到cuda版本。

5. Pytorch官网（网址）

https://pytorch.org/

注：可以点击历史版本，查看老版本torch对应torchvision和torchaudio版本。