Ubuntu下安装tensorflow-gpu版(cuda8+cudnn5)

首先说明下:GPU驱动有很多匹配方案,匹配方案可以是tensorflow 1.5+cuda 9+cudnn7或者tensorflow1.4+cuda 8+cudnn6或者tensorflow1.2+cuda 8+cudnn5,安装时从这些方案选一个即可,这里举例tensorflow1.2+cuda 8+cudnn5方案

1.下载驱动

1下载cuda 8

地址为https://developer.nvidia.com/cuda-80-ga2-download-archive

选项截图为:

 

从中下载得到cuda 8,有时候一会就直接完成了,包的大小不对,那是下载失败了。

2下载cudnn 5 

地址为https://developer.nvidia.com/rdp/cudnn-archive

选项截图为:

 

需要登录账号下载,申请一个并下载下来。

最后下载得到两个文件:


2.开始安装

1安装驱动

Terminel中输入sudo apt-get update更新软件源,

然后在系统设置->软件更新->附加驱动->选择nvidia最新驱动->应用更改,(找到有NVidia选项,选中并apply应用

(2)安装cudn

*首先安装cuda8,安装语句为(针对于run文件):

首先cd 进入run文件放置的目录中(比如我用的 cd /home/fly/Desktop),之后命令输入

chmod u+x cuda_8.0.44_linux.run

sudo ./cuda_8.0.44_linux.run

如果是deb文件,语句是 (针对下载了deb文件)

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb

                    sudo apt-get update

                    sudo apt-get install cuda

*注意在安装时候电脑会特别卡,忍一忍吧,受不了就按按asd键,--More--部分会有点进度显示,等等半小时就行了,不用强制关机。

*在安装中会提示是否接受协议(do you accept the previously read EULA),此时要输入:accept

之后询问 insatall NVIDiA Accelerated Graphics Driver for Linux....,选no

         Insatll the cuda 8.0 toolkit  yes

         Enter toolkit location       选默认的,直接enter过去就行

         Do you want to insatll a symbolic link at ....   yes

        Insatall the cuda 8.0 sample      no

安装好之后会提示 Toolkit:installed in /usr/local/cuda-8.0(之后会按该路径进行配置)

(3)安装cudnn

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/

(4) 配置环境变量

使用语句 sudo gedit ~/.bashrc 进入文件vi .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  

快捷键ctrl+s 保存并退出。

之后命令行输入source.bashrc 是的对文件的修改有效,至此完成配置

5安装TensorFlowGPU版)

*首先 sudo apt-get install python-pip 将pip安装上,

在安装中如果遇到unable to locate package问题,解决方法是重新更换软件源,soft&update->Download from列表中选择other,->在china下列表中找到aliyun表示选中,输入密码->点击close,会提示信息,之后点击Reload,就更新成功了,更新之后再安装即可。

*之后sudo pip install tensorflow-gpu安装上TensorFlow。就可以进入python环境中尝试import tensorflow试试效果了,或者写个计算图,试试加速效果。

*安装太慢的话,可以先在windows电脑下载,下载地址(下载版本根据自己需求,有匹配的方案):https://pypi.python.org/pypi/tensorflow-gpu/1.2.0,下载tensorflow_gpu-1.2.0-cp27-cp27mu-manylinux1_x86_64.whl。

*运行wheel文件的方法为:pip install wheel
之后sudo pip install tensorflow_gpu-1.2.0-cp27-cp27mu-manylinux1_x86_64.whl 即可安装某模块包
*之后直接可以在terminal下运行python环境下,加入以下语句:

出现下图说明gpu安装成功了正在启动加速。

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)





猜你喜欢

转载自blog.csdn.net/maqunfi/article/details/79724947