[Ubuntu18.04]Ubuntu18.04+CUDA9.0+cuDNN7.1.4+tensorflow(gpu)1.8

首先感谢许多网友的分享,对安装作用很大

电脑配置

GPU GeForce GTX 960M

1.CUDA9.0

参考安装的文章为:http://www.zhimengzhe.com/bianchengjiaocheng/qitabiancheng/415560.html
文章只是参考,这一步最关键的还是要看一下官网的安装教程

不同的地方:

我安装的是CUDA9.0,但是换汤不换药,方法一样。
我在安装完系统之后就按照推荐的驱动安装了nvidia390的驱动。
在关闭与开启图形界面的时候,一直结果为...failed, ...not loaded,意思是没有加载,所以也不需要关闭了。

2.cuDNN7.1.4

首先,去下载地址下载cuDNN,注意,需要先注册一个开发者账号。
先解压,解压之后有一个cuda的文件夹

$ tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz

然后复制相应文件,病修改权限

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

因为cuda链接到cuda-9.0文件夹了,所以直接复制到cuda就可以。
这样就安好了。

3.tensorflow-gpu

我一直用的python3,所以用pip3来安装
安装pip3

$ sudo apt install python3-pip

因为默认安装的tensorflow是1.10版本的,但是我想安装1.8的,所以用下面命令

$ pip3 install --upgrade tensorflow-gpu==1.8

这样安装好之后,打开python3
然后导入的时候有错误,Traceback有一长串,但是核心是这一句

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

查到这种问题有三个原因,查看下面文章。
https://blog.csdn.net/weixin_39513374/article/details/80997912
最后,我是通过另一种方法成功的。
检查知道,在目录/usr/local/cuda-9.0/lib64中有文件libcudart.so.9.0
然后执行了以下命令

sudo ldconfig /usr/local/cuda-9.0/lib64

这样导入就没有错误了。
但是在使用过程中,却又出现了这样的错误

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

查到三种解决方法,查看文章:
https://blog.csdn.net/Fourierrr_/article/details/79749899
www.bubuko.com/infodetail-2586137.html
我采用了简单的方法,忽略警告,在每个用tf的程序前,加一个开头

import os
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2'

这样就解决了,如图测试成功。

这样基本的环境就搭好了。

参考

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/
https://www.linuxidc.com/Linux/2017-03/142299.htm
www.zhimengzhe.com/bianchengjiaocheng/qitabiancheng/415560.html
https://ywnz.com/linuxjc/1754.html
http://www.cnblogs.com/platero/p/4118139.html
https://blog.csdn.net/weixin_39513374/article/details/80997912
https://blog.csdn.net/Fourierrr_/article/details/79749899

posted @ 2018-09-20 13:39  aacirq  阅读(309)  评论(0)    收藏  举报