2.安装pytorch

首先安装显卡驱动&＃xff0c;之后安装CUDA与CUDNN,详情可以看一下这个 1.安装tensorflow_Suyuoa的博客-CSDN博客

我当前的CUDA为10.1&＃xff0c;CUDNN为7.6.5

安装之后我们去 https://download.pytorch.org/whl/torch_stable.html 看一下找到我们想要的版本

• 应该先选型再安装&＃xff0c;我这里由于是之前安装了CUDA与CUDNN所以就不想改了

由于我们用的CUDA是10.1,所以我们搜索CU101&＃xff0c;我们选择CU101支持的最新版本1.8.1

点击链接whl下载下来&＃xff0c;我们一会儿用

之后我们用anaconda创建一个虚拟环境

创建成功后是这样的

之后我们激活环境&＃xff0c;之后输入 conda install cudatoolkit&＃61;10.1

然后输入 conda install cudnn&＃61;7.6.5

最后我们切换到刚刚下载链接的目录&＃xff0c;然后输入

之后我们进入运行这样一段代码

如果返回True的话代表pytorch-gpu版安装成功