利用Anaconda配置pytorch。

一,配置cuda以及cudnn(gpu加速)

(如果不没有nvidia显卡的话,只能用CPU,可忽略这一步)

在官网上下载好cuda以及cudnn后,开始配置cuda环境。

二,安装Anaconda

2.1 在官网下载并安装  Anaconda。 

开始菜单中打开Anaconda中的Anaconda Prompt,出现(base),即安装成功,如下:

pytorch安装要多久 pytorch需要装cudnn吗_python


(配置失败请见文章末尾踩坑记录)

2.2 配置环境

安装成功后,停留在上面 Anaconda Prompt 页面上,输入 

conda create -n pytorch python=3.8

 再接着输入y。将所需要的包安装完。

接着,我们输入:

conda activate pytorch

 可以看到

pytorch安装要多久 pytorch需要装cudnn吗_pytorch安装要多久_02

最左边的base,变成了pytorch。就进入了pytorch的编译环境。

2.3 pytorch的安装

在 pytorch官网的对应部分选择我们对应的环境。

pytorch安装要多久 pytorch需要装cudnn吗_python_03

Pytorch Build:一般选择Stable(稳定版)。

Your OS:PC的操作系统。

Package:我们是在anaconda上安装,所以选Conda。

Language:选择Python。

Compute Platform:这里我们要根据我们在步骤一中配置的cuda的版本来选择。

win+r,输入cmd调出命令行窗口。输入

nvcc -V
nvidia-smi

都可查看当前cuda版本。

pytorch安装要多久 pytorch需要装cudnn吗_pytorch安装要多久_04

在最下面一行(Run this Command)后复制对应的命令。

对于上图即:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

完成之后,我们在Anaconda Powershell prompt 窗口中复制上述命令。

pytorch安装要多久 pytorch需要装cudnn吗_官网_05

再接着输入y。等待即可安装成功。

(速度慢、失败见踩坑记录)

2.4 验证环境安装 

      关闭Anaconda Powershell prompt,再重新打开,再次通过命令conda activate pytorch 进入pytorch环境。

依次输入:

python

 

pytorch安装要多久 pytorch需要装cudnn吗_pytorch安装要多久_06

import torch

pytorch安装要多久 pytorch需要装cudnn吗_python_07

torch.cuda.is_available()

pytorch安装要多久 pytorch需要装cudnn吗_官网_08

 可见,返回Ture,说明cuda安装成功,否则pytorch只能用cpu运行。

三,踩坑记录

3.1 上文2.1部分踩坑

 这里如果产生类似报错:

pytorch安装要多久 pytorch需要装cudnn吗_pytorch安装要多久_09

 请检查系统环境变量中,Path路径。所有的变量必须分条写,不能用“;”写在一条里。

pytorch安装要多久 pytorch需要装cudnn吗_python_10

 

3.2 上文2.3中遇到的安装失败、速度太慢等问题。

(1)看了很多博客建议用清华镜像,但是我试了很多遍都失败了。第一次很快的安装上了CPU版本的,导致电脑重装从头再来。这里还是建议就在官网下载,我是晚上放着,早上起来就装好了。多试几次。

(2) 安装的时候不要挂VPN,必然失败,或者报错。我因为这个浪费了一个下午。

(3)更新conda 命令。

3.3 torch.cuda.is_available()返回false可能的原因

(1)检查是否有GPU。

(2)检查CUDA以及CUDNN环境、系统Path变量配置。

(3)检查显卡驱动是不是最新的。