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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值