PytorchGPU版本

conda create -n myenv python=3.8

conda activate myenv

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --timeout 1000

pip list

import torch
 
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())  #输出为True，则安装成功