安装PyTorch

PyTorch一分为三：torch、torchvision、torchaudio

 

#pip下载命令

pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

 

在conda中使用python解释器,

import torch检测是否成功

torch.cuda.is_available() 返回True表示成功