一、安装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秒,之后就顺利跳过,成功开机。