相关链接:
安装NVIDIA驱动步骤
安装tensorflow-gpu步骤
安装CUDA-9.0
若不对gcc进行降级,会出现以下错误:
(注:高版本可能就不需要降级了,大家以后遇到这样的报错就再降级也可以)
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
官网下载CUDA
注意:tensorflow对CUDA的版本有要求,见https://github.com/mind/wheels
到官网https://developer.nvidia.com/cuda-toolkit-archive下载cuda
根据需求安装适宜版本
下载文件的列表:
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
配置环境变量
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}}
激活
source ~/.bashrc
CUDA测试
gcc g++调至低版本
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
cd ../bandwidthTest
sudo make
./bandwidthTest
测试成功都会出现
Result = PASS
查看cuda版本:
cat /usr/local/cuda/version.txt
安装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/lib64/
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
安装完毕
安装完成后可用 nvcc -V 命令验证是否安装成功,若出现以下信息则表示安装成功:
yhao@yhao-X550VB:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61
但可能会找不到nvcc指令,还有一种方式:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
其他方法记载:https://blog.csdn.net/daopper/article/details/81206673针对ubuntu16版本