Ubuntu 18.04 搭建PyTorch深度学习环境

一、安装Ubuntu 18.04

二、安装NVIDIA驱动、CUDA、cuDNN

1. 下载CUDA和cuDNN

到NVIDIA官网下载对应版本的CUDA，选择runfile格式

https://developer.nvidia.com/cuda-downloads

到NVIDIA官网下载对应版本的cuDNN，登录，下载cuDNN Library for Linux

https://developer.nvidia.com/cudnn

2. 安装驱动、CUDA和cuDNN

（1）禁用nouveau驱动

编辑文件blacklist.conf

sudo gedit /etc/modprobe.d/blacklist.conf

在文件最后部分插入以下两行内容，保存

blacklist nouveau
options nouveau modeset=0

更新系统、重启

sudo update-initramfs -u
reboot

（2）安装gcc和ubuntu-make

sudo apt-get update
sudo apt-get install gcc
sudo apt-get install ubuntu-make

（3）同时安装显卡驱动和CUDA

sudo sh cuda_xxxxxx.run

（4）安装cuDNN

解压，进入解压目录

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

三、安装Anaconda、PyTorch

1. 安装Anaconda

到官网下载Anaconda

https://www.anaconda.com/distribution/

运行安装

bash Anaconda3-xxxxxx.sh

2. 安装PyTorch

到官网下载PyTorch

https://pytorch.org/

重启终端，运行安装

pip install torch-xxxxxx.whl
pip install torchvision-xxxxxx.whl