一．安装显卡驱动
显卡驱动官网：https://www.nvidia.cn/geforce/drivers/
或者：https://www.nvidia.cn/Download/Find.aspx?lang=cn
目前安装：456.38
二．安装显卡驱动支持的cuda
打开显示驱动控制面板看其支持的CUDA版本
目前安装：CUDA 11.1.70
三．安装相应的pytorch
官网进行相应的选择
目前安装：pytorch1.6.0 cuda 10.2
四．测试代码
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
测试结果
1.6.0
10.2
True
五．Pycharm设置anaconda的环境
参考：https://blog.csdn.net/zfcjhdq/article/details/104919067
六．在anaconda环境中新建一个环境，重新安装pytorch再测试