ubuntu 18.04 安装显卡驱动+CUDA9.0+cuDNN+pytorch1.0预览版

最近出了pytorch1.0的预览版（nightly版本），好久没用pytorch了，装一个耍耍。。。
安装条件：显卡支持cuda且驱动可以安装384以上版本

Ubuntu版本升级

如果你的版本是Ubuntu 16 的话可以选择命令升级：

第一：升级到最新的Ubuntu16.04 LTS

sudo apt-get update
sudo apt-get upgrade --yes
sudo apt-get dist-upgrade --yes

第二：升级到Ubuntu18.04

sudo do-release-upgrade
#如果不行的话，输入如下
sudo do-release-upgrade -d

第三：软件配置

在升级过过程中会有一些原先安装的软件的配置询问，都选着原来的配置即可。。

第四：重启

重启过后则升级完成。

NVIDIA显卡驱动安装

第一：检查显卡和推荐驱动：

ubuntu-drivers devices

结果如下：
在这里插入图片描述
可以看出我的笔记本显卡是 GTX 965M ，推荐的驱动是 390

如果没有你想安装的驱动可以添加NVIDIA的PPA：

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

再查看匹配驱动：
在这里插入图片描述

多出了其他显卡推荐，而且这里推荐我安装410版本，因为安装cuda9.0，要驱动版本在384以上，所以我的390就可以，但是你要是使用cuda10的话就需要安装410版本，因为pytorch官网上没有放出支持cuda10的版本，我就选择安装cuda9.0，应该能稳定一些吧。。

第二：驱动安装

自动安装推荐驱动

sudo ubuntu-drivers autoinstall

也可以选择其中一个：

sudo apt install nvidia-390

安装完成后重新启动。。
检验是否安装成功：

nvidia-smi

出现如下内容，说明安装成功：
在这里插入图片描述

安装CUDA 9.0

第一：下载安装包

你可以选着取官网下载，下载地址

也可以在百度云盘上下载：
链接: https://pan.baidu.com/s/1QxZUHaAZcG59sApsD_7lkg 提取码: xb5m
最大的那个是安装包，其他都是升级包。。

第二：gcc降级

由于CUDA 9.0仅支持GCC 5.3及以下版本，而Ubuntu 18.04预装GCC版本为7.3，
故手动进行降级：
下载安装 gcc4.8

sudo apt-get install gcc-4.8
sudo apt-get install g++-4.8

配置：

cd /usr/bin
sudo mv gcc gcc.bak #备份
sudo ln -s gcc-4.8 gcc #重新链接
sudo mv g++ g++.bak
sudo ln -s g++-4.8 g++

查看版本号：
gcc -v && g++ -v 　　　　　　　　　　　　
显示版本皆为4.8，则说明成功。

安装过程：

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

安装完成后，需要配置cuda环境：

sudo vim ~/.bashrc
在最后添加：

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export CUDA_HOME=/usr/local/cuda-9.0

重新加载一下：
source ~/.bashrc
查看cuda是否成功安装：

$nvcc -V
# 出现如下内容说明安装成功
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176

安装CuDNN

第一：下载

可以选择官网下载：https://developer.nvidia.com/rdp/form/cudnn-download-survey但是需要注册
也可以选择在百度网盘上下载：
链接: https://pan.baidu.com/s/19_kwsEKrj65Zm_DN2G_avw 提取码: 71m3

第二：安装

解压后，将文件复制到cuda相应文件加就好：

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   /usr/local/cuda/lib64/libcudnn*

这样CuDNN 安装完成。。

安装pytorch 1.0 预览版

pytorch官网上可以查到如何安装，网站地址：https://pytorch.org/get-started/locally/
推荐使用conda安装，当然你需要先安装anaconda

使用如下命令即可以安装：

conda install pytorch-nightly -c pytorch

但是国内速度很慢，可以使用国外主机下载然后传到国内主机上，再进行本地安装。这有一个我上传到百度上的：
链接: https://pan.baidu.com/s/1D6NkSVbStTuaNuay_m6b5w 提取码: g954

下载下来之后把这个安装包放到anaconda存放包的目录下，如:
~/anaconda3/pkgs/ 这个目录下
然后执行命令：

conda install --use-local  pytorch-nightly-1.0.0.dev20181010-py3.6_cuda9.0.176_cudnn7.1.2_0.tar.bz2

这样就可以成功安装。。

检验是否安装成功：
python 打开python
输入：

import torch

不报错说明安装成功。
输入：

>>> torch.cuda.is_available()
True

返回True说明cuda可以正常使用，则全部安装完成。。。。
在这里插入图片描述