下载pytorch:https://pytorch.org/get-started/locally/
选择对应版本并下载
这里我的CUDA版本是10.1,执行代码
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
新建一个test.py文件,或是直接在命令行中输入以下代码
#python 3
import torch
print(torch.cuda.is_available())
返回True,证明安装成功