首先安利一个ubuntu 16.04来截图的快捷方式:
进入keyboard,之后点击Custom Shortcuts。点击‘+’号,在跳出的窗口中,依次输入“截图”和“ gnome-screenshot -a ”,效果图如下:
之后就可以按快捷键去截图了。
进入正题:
在安装好cuda 8.0 以后,接下来博主我就开始安装cudnn 5.1了,首先我们去官网下载cudnn 5.1的压缩包
网址在这里:cudnn5.1的下载地址(必须要注册以后,才可以下载)
下载完成后,我们就可以开始:
首先进入到我们下载那个文件的地址:这里是:Downloads
查看cudnn 5.1版本:
为了检查cudnn5.1有没有安装好,输入 :cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A 2
如下:
至此,我们的ubuntu 16.04 的显卡驱动、cuda 8.0 、cudnn 5.1上面的内容已经配置好了。这里我们开始去安装tensorflow。
步骤:
1、首先我们先检查我们的显卡驱动、cuda8.0、cudnn 5.1是否安装好了,在前面的blog中我们可以了解,这里我再次写出来:
博主的显卡安装的链接:显卡驱动安装教程
方法(1)nvidia-settings,效果图如下:
方法(2) lspci | grep -i nvidia,效果如下:
方法(3)nvidia-smi,效果如下:
上面的任何一种就行了,如果和博主一样,那么你的显卡驱动装好了。
2、cuda8.0 的安装是否完成,参考博主的上篇blog,有具体的步骤:cuda 8.0(安装和检查)链接
3、检查我们的cudnn 5.1有没有安装好:见该篇blog前面部分,有详细的解析。
上面的步骤1,2,3完成后,我们可以开始安装tensorflow(GPU版本)了。
首先:
sudo apt-get install python3-pip
sudo pip3 install tensorflow-gpu
我们会发现会报错,原因是安装的tensorflow 1.3.0版本与cuda8.0 + cudnn5.1环境不兼容,在python中import后会出现各种错误。
解决方案一:要安装1.2.0或1.2.1版本:
sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.2.0-cp35-cp35m-linux_x86_64.whl
这里使用的tsinghua的镜像,效果如下:
现在测试是否能用:
python3
> import tensorflow
>
解决方案2:你就需要使用cuda 9.0以上,才能使用这个版本的tensorflow