2.安装pytorch

首先安装显卡驱动，之后安装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版安装成功