安装CUDA-9.0

注意：tensorflow对CUDA的版本有要求，见https://github.com/mind/wheels
到官网下载cuda9.0
在这里插入图片描述

下载文件：

    cuda_9.0.176_384.81_linux.run
    cuda_9.0.176.1_linux.run
    cuda_9.0.176.2_linux.run
    cuda_9.0.176.3_linux.run
    cuda_9.0.176.4_linux.run

CUDA安装

sudo sh cuda_9.0.176_384.81_linux.run

安装过程中，除了安装NVIDIA显卡的部分选no，其他部分都可以选yes
在这里插入图片描述

安装补丁

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 sh cuda_9.0.176.4_linux.run

gcc g++降级

测试cuda需要对gcc，g++版本进行降级(降至6.0以下)
ubuntu18.04自带gcc g++ 7.30
查询方式：

gcc --version
g++ --version

下载低版本gcc g++，并进行设置

sudo apt install gcc-4.8 g++-4.8 
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 50 
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 50   
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 70 
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 70

这样就可以通过以下命令来切换使用的gcc和g++版本:

sudo update-alternatives --config g++

在这里插入图片描述

sudo update-alternatives --config gcc

在这里插入图片描述

配置环境变量

sudo gedit ~/.bashrc

在文档结尾添加：

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

CUDA测试

gcc g++调至低版本

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

    cd ../bandwidthTest
    sudo make
    ./bandwidthTest

测试成功都会出现

Result = PASS

安装cudnn

到官网下载，本人下载的是cudnn7.0.5 for cuda 9.0

tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
sudo cp  cuda/include/cudnn.h /usr/local/cuda-9.0/include/
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安装完毕

（有些教程还有以下步骤https://blog.csdn.net/xierhacker/article/details/53035989）

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7 #删除原有动态文件
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7 #生成软衔接
sudo ln -s libcudnn.so.7 libcudnn.so #生成软链接

其他方法记载：https://blog.csdn.net/daopper/article/details/81206673针对ubuntu16版本
下载CUDNN 7.1 for CUDA 9.0（共三个文件）deb类型
安装：

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

验证：

cp -r /usr/src/cudnn_samples_v7/ ~
cd ~/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

安装NVIDIA驱动步骤
 安装tensorflow-gpu步骤

真实机下ubuntu18.04安装NVIDIA驱动，CUDA-9.0，CUDNN-7.05，tensorflow过程(二)