Ubuntu 16.04 安装tensorflow-gpu、cuda和cuDNN
python3.5+tensorflow-gpu1.5 + cuda9.0 + cudnn9.0
禁用开源nouveau驱动
sudo vim /etc/modprobe.d/disable-nouveau.conf
在最后添加
blacklist nouveau
options nouveau modeset=0
然后执行下面的命令,使配置生效
sudo ldconfig
sudo update-initramfs -u
验证是否成功禁用nouveau
sudo lspci | grep nouveauinstall
若没有输出则禁用成功。
安装Linux4.4.0内核
sudo apt-get install linux-headers-4.4.0-93-generic
sudo apt-get install inux-image-4.4.0-93-generic
重启计算机,在启动菜单进入Ubuntu高级选项,选择linux 4.4.0内核启动系统:
在登录界面按ctrl+alt+f1进入tty1模式,关闭桌面:
sudo service lightdm stop
安装cuda9.0:
sudo ./cuda_9.0.*_linux.run
安装cuDNN9.0:
切换至cuDNN Tar 文件所在目录解压
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
复制以下文件到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
环境变量配置:
sudo vi ~/.bashrc(控制台有效) 或者 /etc/profile
将以下内容写到文件尾部。
export PATH=$PATH:/usr/local/cuda-9.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
保存后,更新文件
source ~/.bashrc
重启,同样在登录界面按ctrl + alt + f1进入tty1,安装NVIDIA显卡驱动:
sudo sh ./NVIDIA-Linux-x86_64-38* -no-x-check -no-nouveau-check -no-opengl-files
重启,进入图形界面:
安装tensorflow:
pip3 install tensorflow-gpu==1.5
测试是否安装成功:
python3
import numpy import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))
exit()