首先安装显卡驱动，之后安装CUDA与CUDNN,详情可以看一下这个 1.安装tensorflow_Suyuoa的博客-CSDN博客
我当前的CUDA为10.1，CUDNN为7.6.5
安装之后我们去 https://download.pytorch.org/whl/torch_stable.html 看一下找到我们想要的版本
- 应该先选型再安装，我这里由于是之前安装了CUDA与CUDNN所以就不想改了
由于我们用的CUDA是10.1,所以我们搜索CU101，我们选择CU101支持的最新版本1.8.1
点击链接whl下载下来，我们一会儿用
之后我们用anaconda创建一个虚拟环境
创建成功后是这样的
之后我们激活环境，之后输入 conda install cudatoolkit=10.1
然后输入 conda install cudnn=7.6.5
最后我们切换到刚刚下载链接的目录，然后输入
之后我们进入运行这样一段代码
如果返回True的话代表pytorch-gpu版安装成功
