1、首先安装anaconda
2、创建虚拟环境EnvTorch
3、安装cuda版本
4、根据cuda(通过nvidia-smi命令查看)版本选择执行不同的安装命令:(此处是使用cuda 12.1版本来安装)
conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=12.1 -c pytorch -c nvidia
5、执行python代码验证
import torch
print(torch.__version__)
print(torch.cuda.is_available())
查看是否可以使用cuda