Ubuntu配置GPU版本pytorch环境（含NVIDIA驱动+Cuda+Cudnn）

转自：https://blog.youkuaiyun.com/gdymind/article/details/82153643

本文更新于2018年8月底

概述
步骤如下： 
1. 安装Ubuntu 
2. 安装NVIDIA 显卡驱动 
2. 安装NVIDIA Cuda 
3. 安装NVIDIA CuDNN 
4. 安装GPU版本的PyTorch

安装Ubuntu
系统版本选择
之前pytorch是不支持Windows的，现在貌似是支持了，但我们还是选择Ubuntu。注意Ubuntu的版本要与cuda兼容，目前官网支持的版本有17.10和16.04，笔者使用了Ubuntu 16.04版。

安装过程略，注意安装完系统后更改/etc/apt/sources.list文件换成国内源，更换方法可以搜索“ubuntu16.04 国内源”。如果换源后sudo apt update还是失败，就去检查/etc/apt/sources.list.d/目录中的文件是否含有需要翻墙才能上的网站（比如安装了chrome就会有这个问题==）。

安装NVIDIA显卡驱动
安装NVIDIA驱动有多种方式，比如可以去官网下载最新版，这里介绍最简单的一种：

打开System Settings… 

选择System下的Software & Updates 

在弹出的窗口中选择Additional Drivers选项卡 

选择NVIDIA的显卡驱动，点击Apply changes就可以等驱动安装完成啦，安装完成后重启一下～
安装NVIDIA Cuda
到官网下载文件：选择对应版本（见下图）并点击Download按钮： 

等下载完成后，依次执行下面的命令，注意<version>要替换成对应的版本号，如9-2 

安装NVIDIA CuDNN
1.到官网下载文件：点击如图按钮。下载需要NVIDIA的账号，没有的需要先注册一个 

选择Runtime Library版的deb文件进行下载： 

安装cudnn过程与cuda类似：

执行命令sudo dpkg -i <name>，其中<name>为刚才下载的deb文件名
执行命令sudo apt install <name>,其中<name>要和自己下载的cudnn版本匹配，比如这里是libcudnn7
安装GPU版本的PyTorch
这里选择用pip进行安装，首先需要安装pip：执行命令sudo apt intall python-pip3

现在建议配置pip虚拟环境，为此我们需要配置virtualenv（有关virtualenv的详细介绍可以看看廖雪峰Python3教程和官方documentation）：

安装virtualenv：执行命令pip3 install virtualenv
新建一个虚拟环境（这里取名为env-pytorch)：在终端中切换到想要新建env-pytorch的目录下（创建完虚拟环境后此目录下会多一个名为env-pytorch的文件夹，包含该虚拟环境相关的全部文件），执行命令virtualenv --no-site-packages venv -p <PYTHON_EXE>，其中<PYTHON_EXE>是Python的版本号，例如python3.6
到官网选择对应版本之后会得到相应的安装命令，如下图：

为了验证以上安装成功，在终端输入python进入Python交互模式，然后依次输入：
import torch
torch.cuda.is_available()
1
2
如果返回结果是True，证明安装成功了
--------------------- 
作者：gdymind 
来源：优快云 
原文：https://blog.youkuaiyun.com/gdymind/article/details/82153643 
版权声明：本文为博主原创文章，转载请附上博文链接！