快速安装CUDA版本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

4. 查看本机cuda版本

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

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

5. Pytorch官网（网址）

https://pytorch.org/