安装 PyTorch: conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

检测GPU是否可用方法

输入命令

import torch
print(torch.__version__)

torch.cuda.is_available()

返回True

参考

PyTorch documentation