根据电脑cuda版本安装支持gpu的torch

先nvcc -V查看cuda版本，如12.3
下载<=该版本的CUDA: https://developer.nvidia.com/cuda-toolkit-archive
下载三件套: torch, torchvision, torchaudio：https://download.pytorch.org/whl/torch_stable.html
pip install ___.whl 即可安装

查看版本

import torch
print("PyTorch version:", torch.__version__)
print("CUDA available:", torch.cuda.is_available())
print("CUDA version:", torch.version.cuda)

PyTorch version: 2.3.1+cu121
CUDA available: True
CUDA version: 12.1