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 nouveauoptions 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

 

重启,进入图形界面:

测试是否安装成功

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()

 

 

猜你喜欢

转载自www.cnblogs.com/liuwei0773/p/9334465.html