安装pytorch

查看NVIDIA的CUDA Version

在命令行输入命令nvidia-smi

cuda是向下兼容的，比如12.2版本可以兼容11.8、12.1

查看Pytorch支持的CUDA 版本

可以点击这个链接进行查看Pytorch支持的版本 Start Locally | PyTorch

这一步的cuda版本一定不能高于上一步查出来的CUDA Version

下载CUDA Toolkit

CUDA Toolkit 12.3 Update 1 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cuda-downloads版本一定要各pytorch一致，比如上面的是12.3的版本，那就可以滚动到下面查看已经归档的版本

安装pytorch的三个套件

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

验证pytorch是否安装成功

import torch

print(torch.cuda.is_available())