本机配置参考第一篇文章https://blog.csdn.net/qq_23612831/article/details/88289943
1、先安装和更新各个包
sudo apt-get update
sudo apt-get install python-pip python-dev
sudo pip install --upgrade pip
2、安装NVIDIA驱动
NVIDIA驱动过程安装略过
附:CUDA与NVIDIA驱动对应关系:
CUDA | 9.1 | 9.0 | 8.0 | 7.5 | 7.0 |
---|---|---|---|---|---|
NVIDIA | 387.xx | 384.xx | 367.4x | 352.xx | 346.xx |
本机为NVIDIA 384.130,故可以安装CUDA9.0
附:降gcc版本方法(CUDA8.0支持gcc5.0以下,CUDA9.0支持gcc6.0以下)
降gcc到5.0以下方法:
sudo apt-get install g++ -4.9
3、依赖库安装(CUDA9.0)
sudo apt-get install freeglut3-dev build-essential libxll-dev libxmu-dev
sudo apt-get install libxi-dev libgll-mesa-glx libglul-mesa libglul-mesa-dev
4、下载CUDA9.0相关
下载CUDA9.0 (linux—x86_64—Ubuntu—16.04—deb)
下载Base Install 以及其几个patch
5、将安装包放在自己指定目录
安装sudo dpkg -i cuda-rep-ubutu1604-9-0xxxxx.deb
(省略安装文件名字)
中途需要按要求输入key指令:sudo apt-key add /var/……
(安装过程有提示,省略)
安装几个patchsudo dpkg -i xxxxx
(省略安装包名称)
sudo apt-get update
sudo apt-get install cuda
此时,可见,在目录 usr/local 下,已经有cuda,cuda-9.0两个文件夹
6、环境配置
注:用户安环境设置: sudo gedit ~/.bashrc
让设置生效 :source ~/.bashrc
系统环境设置:sudo gedit /etc/profile
让设置生效 :source /etc/profile
本次安装,设置的是系统环境:sudo gedit /etc/profile
在打开的文件末尾添加:
export PATH = /usr/local/cuda-9.0/bin:$PATH
export CUDA_HOME = /usr/local/cuda-9.0
export LD_LIBRARY_PATH = /usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH = /usr/local/cuda-9.0/extras/CUPT/lib64:$LD_LIBRARY_PATH
保存退出以后,再次执行source /etc/profile
7、测试CUDA安装成功与否
方法1:
cd /usr/local/cuda9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
若有GPU信息,则安装成功(最后会有一个Result = PASS)
方法2:
cd /usr/local/cuda9.0/samples/5_Simulations/fluidsGL
sudo make
./fluidsGL
如果最后出现一个动态的花里胡哨的界面,则证明CUDA安装成功
8、cuDnn 7.0 安装
下载:cudnn v7.0.5 for CUDA 9.0
cudnn v7.0.5 Library for linux
将下载的包放到自己指定的目录
在目录下运行:tar -xvzf cudnn-9.0-linux-x64-v7.solitairetheme8
(后面为自己下载的包名)
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*
(注:将复制的目标目录的cuda更改为cuda-9.0也一样,若上述测试不成功,可以更改为cuda9.0,例:sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
)
测试安装成功与否:nvcc-V
若结果显示相关信息,则表明测试成功。
9、安装TensorFlow
安装前,先安装:sudo apt-get install libcupti-dev
网上下载好TensorFlow对应版本,本次使用tensorflow_gpu_1.10_cp27_cp27mu_manylinuxl_x86_x64.whl
安装:sudo -H pip install xxxx.whl
(xxx为对应安装包)
注:若提示pip是家目录的,则用sudo -H ,没提示,则可以不用 -H
测试:
python
import tensorflow as tf
hello = tf.constant('hello world')
sess = tf.session()
print = (sess.run(hello))
若结果打印出“hello world”,则表示安装成功。