您的位置:首页 > 其它

ubuntu16.04配置cuda8.0+cudnn5.1+tensorflow1.2

2017-11-25 23:56 651 查看
捣鼓了一天,中午把烦人的环境给配好了,这是第二次配置了,呜呜呜,还是遇到很多问题,于是决定写篇博客记录一下,万一以后还要装呢T-T

大部分都是参照网上教程来的,但是还是走了不少弯路。

(1)安装ubuntu16.04啦,不多说,上连接

https://jingyan.baidu.com/article/3c48dd348bc005e10be358eb.html

**

(2)安装nvidia驱动

**

首先,要去nvidia官网找到自己的型号对应的显卡驱动,记住编号,安装要用到

然后,禁用系统默认的集成驱动

Ubuntu系统集成的显卡驱动程序是nouveau,它是第三方为NVIDIA开发的开源驱动,我们需要先将其屏蔽才能安装NVIDIA官方驱动。

将驱动添加到黑名单blacklist.conf中,但是由于该文件的属性不允许修改。所以需要先修改文件属性。

查看属性

$sudo ls -lh /etc/modprobe.d/blacklist.conf


修改属性

$sudo chmod 666 /etc/modprobe.d/blacklist.conf


用gedit打开

$sudo gedit /etc/modprobe.d/blacklist.conf


在该文件后添加以下几行:

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb


准备工作做完之后,就可以开始安装啦

Ctrl+Alt+F1进入命令行模式,关闭图形系统

$sudo service lightdm stop


安装N卡驱动程序(我的显卡推荐的是nvidia-384,就是上面找的编号,自己对应好就好了)

$sudo apt-get install nvidia-384


安装完成后,启动图形系统

$sudo service lightdm start


上面的命令执行后会自动转到图形界面,因为之前Ubuntu系统集成的显卡驱动程序nouveau被禁用了,这时候可能无法显示图形界面,此时再按下Ctrl+Alt+F1进入命令行模式,输入reboot 重启计算机即可。

(3)cuda安装

去官网找到自己的版本下载,按照官网的指导,

sudo sh cuda_8.0.27_linux.run




然后会出现一个神奇的界面,需要一直按enter键,除了nvidia显卡选项必须选择no(这个我们之前装过了)其他都按照默认来。

安装完毕后,再声明一下环境变量,并将其写入到 ~/.bashrc 的尾部:

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


然后设置环境变量和动态链接库,在命令行输入:

$ sudo gedit /etc/profile


在打开的文件末尾加入:

export PATH=/usr/local/cuda/bin:$PATH


保存之后,创建链接文件:

sudo gedit /etc/ld.so.conf.d/cuda.conf


在打开的文件中添加如下语句:

/usr/local/cuda/lib64


然后执行

sudo ldconfig


使链接立即生效。

测试cuda的Samples

cd /usr/local/cuda-7.5/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery


如果显示的是一些关于GPU的信息,则说明安装成功了。

(4)cudnn5.1安装

安装cudnn–> cuda

下载包:cudnn-8.0-linux-x64-v5.1.tgz 解压后得到 cuda 文件夹

在cuda 目录下执行如下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/


配置环境变量:gedit .bashrc

添加如下路径:

`

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH

export CUDA_HOME=/usr/local/cuda-8.0

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


保存退出 , 执行:source .bashrc 使配置文件生效

(5)安装tensorflow

sudo apt-get install python3-pip python3-dev # for Python 3.n


去清华tensorflow找安装镜像

https://mirror.tuna.tsinghua.edu.cn/help/tensorflow/

选择自己匹配的tensorflow版本号就好了。官网的总是下载不下来,很是烦躁,还好有清华镜像。

到此完美结束,撒花~~

传送门:

送上cudnn5.1+cuda8.0

链接: https://pan.baidu.com/s/1nvOgBkX 密码: 8nsg

文末理一理这些软件的关系:

cuda9不支持Tensor flow,所以像我一样的喜欢用最新版本的童鞋注意啦

其次cudnn6支持的是tensorflow1.3及以上

cudnn5支持的是1.2,以下支不支持就靠大家自己实践了。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: