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

2.安装pytorch_第1张图片

点击链接whl下载下来,我们一会儿用

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

2.安装pytorch_第2张图片

创建成功后是这样的

2.安装pytorch_第3张图片

之后我们激活环境,之后输入 conda install cudatoolkit=10.1

2.安装pytorch_第4张图片

然后输入 conda install cudnn=7.6.5

2.安装pytorch_第5张图片

最后我们切换到刚刚下载链接的目录,然后输入

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

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

你可能感兴趣的:(pytorch笔记,pytorch,深度学习,人工智能)