ubuntu16.04安装cuda8.0+cuDNN6.0+tensorflow1.4配置过程

一、安装cuda8.0
1、下载cuda_8.0.61_375.26_linux.run
2、进入存放cuda_8.0.61_375.26_linux.run的文件夹

sudo sh cuda_8.0.61_375.26_linux.run

在安装时，由于事先已经安装了Nvidia的驱动，所以，第一项选择不安装nvidia驱动，其他到都是yes和默认，切记！选择不安装nvidia驱动
之后会显示有部分没安装，没关系。

3、

sudo gedit /etc/profile

打开“profile”文件，在末尾处添加（注意不要有空格，不然会报错):

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH

注意，千万不要乱修改.bashrc，否则会导致开不了机！

4、重启

sudo reboot

（有许多教程说在安装cuda8.0之前需要将GCC降低版本，但是我没有降低版本也成功安装了，本人ubuntu版本为16.04.2，gcc版本为5.4。引用别人原文：“ubuntu16.04的gcc版本是5.4，而cuda8.0在该系统中支持的gcc版本不超过5.3，所以第一步需要对GCC版本进行降级。”）

5、测试样例

cd NVIDIA_CUDA-8.0_Samples
sudo make

等待20分钟

6、

cd NVIDIA_CUDA-8.0_Samples/bin/x86_64/linux/release/
./deviceQuery

如果输出GPU信息，则安装成功。

二、安装cuDNN6.0
1、下载cudnn-8.0-linux-x64-v6.0.tgz
解压，进入include文件夹

sudo cp cudnn.h /usr/local/cuda/include/  #复制头文件

2、进入lib64文件夹，执行

sudo cp lib* /usr/local/cuda/lib64/  #复制动态链接库

3、执行

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

三、安装tensorflow_gpu1.4

sudo apt-get install libcupti-dev
sudo apt-get install python-pip python-dev
sudo pip install tensorflow-gpu

四、跑个测试程序

import tensorflow
hello = tf.constant('hello,Tensorflow')
sess = tf.Session()
print sess.run(hello)

五、安装之后，如果重启发现计算机卡在紫屏那里，解决方法如下：

1、选择ubuntu16.04，进入ubuntu的引导的界面，按“e”，进入grub文本模式
2、在Splash这个单词后面加空格，输入nomodeset，然后按F10。【注】Splash这个单词在屏幕中下部，需要按光标下键才可以看到的。
3、顺利进入系统
4、永久改变grub：

cd /etc/default
cp grub ~/        #备份一下grub
sudo gedit grub

修改

GRUB_CMDLINE_LINUX_DEFAULT="quit splash = nomodeset "

保存【注意：等号两边一定要有空格！！！！】

sudo update-grub       #更新grub

之后重启，紫屏大概存在30秒，之后就顺利跳过，成功开机。

ubuntu16.04安装cuda8.0+cuDNN6.0+tensorflow1.4配置过程

猜你喜欢