一、conda创建虚拟环境
python和pytorch版本对应
conda create --name myenv python=3.8
二、安装对应cuda版本的pytorch-gpu
不同的cuda版本对应不同的pytorch,我这里安装的是cuda11.8
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
三、在python环境中查看pytorch-gpu是否安装成功
import torch
print(torch.cuda.is_available())