[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
 
                    
                     
                    
                 
                    
                
 
                
            
         
         浙公网安备 33010602011771号
浙公网安备 33010602011771号