Ubuntu18 安装 CUDA9.0 和 cuDNN 史

安装之前查看了大量文章，最终一次就好，故将每一步记录下来。

更多福利以及文章，详见 -> 点击这里

一、显卡配置

GeForce 940MX

二、安装显卡驱动(重中之重)

在 官网 选择显卡的参数，下载合适的驱动。

重启，输入 nvidia-msi ，如下图所示，则驱动安装完毕。

三、下载

1.下载CUDA9.0

虽然已经有Ubuntu18.04支持的10.0版本CUDA，但是博主的显卡太差，所以安装9.0版本。下载CUDA9.0 。

将下面这几个全部下载下来。

2. 下载cuDNN

注意：cuDNN要和CUDA的版本对应，安装第5,6,7三个 deb包，需要注册一下。 下载cuDNN v7.1.3 。

四、安装

1. gcc版本问题

Ubuntu18.04预装GCC7.3，而CUDA9.0支持GCC6.0以下版本。

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

2. 安装 TensorFlow-gpu

sudo pip3 install tensorflow-gpu

3. 安装CUDA9.0

此时应该已经有四个cuda_9.0********.run文件和三个libcudnn7*******.deb文件。

运行 sudo sh cuda_9.0.176_384.81_linux.run ，出现下图所示：

按 Ctrl + c，按照提示输入，注意 第三个选项问是否安装NVIDIA Accelerated Graphics Driver驱动，选否：

sudo sh cuda_9.0.176.1_linux.run
sudo sh cuda_9.0.176.2_linux.run
sudo sh cuda_9.0.176.3_linux.run

添加环境变量

sudo gedit ~/.bashrc

在末尾添加：
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-9.0

测试CUDA

我在装的时候出现了Warning提示，然后就测试了一下：

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery

显示 PASS，安装成功。

4. 安装cuDNN

sudo dpkg -i libcudnn7_7.1.3.16-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.1.3.16-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.1.3.16-1+cuda9.0_amd64.deb

五、测试

1.

我的 Home 文件夹下有一个 dpkg 后出现的 NVIDIA_CUDA-9.0_Samples 文件夹，

/home/cai/NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

如下所示，出现GPU的信息，表示安装成功：（请忽略博主的辣鸡显卡）

2.

运行代码：

import tensorflow as tf
tf.enable_eager_execution()
hello = tf.constant('hello world')
print(hello)

若出现如下所示GPU的信息，说明安装成功：

六、卸载(如果需要)

cd /usr/local/cuda/bin
sudo ./uninstall_cuda_9.0.pl
sudo rm -rf cuda-9.0