电脑配置如下:
安装过程中使用的安装包:
补充
https://developer.nvidia.com/cuda-downloads
https://www.continuum.io/downloads
https://github.com/bazelbuild/bazel/releases
本文分为两个大部分来介绍,首先介绍cuda与cudnn的安装,然后介绍使用源码安装tensorflow。
一、安装cuda与cudnn
1、Nvidia驱动安装-run文件安装
下载完名称为NVIDIA-Linux-x86_64-396.18.run的文件。使用以下命令:
加入以下语句,将nouveau禁止命令写入文件。
调用指令禁止nouveau。
建立新的内核并重启。
使用ctrl + alt+ F1进入文本模式,关闭x server。
切换NVIDIA安装包指定目录,赋予权限并进行安装
返回图形界面
检查驱动是否安装成功,出现图片所示,驱动就安装完成了。
2、安装cuda
安装的CUDA 可以到官网去下载,我在此安装cuda_9.1.85_387.26_linux.run。执行以下命令:
按q结束cuda的描述,然后输入accept,在提示是否安装NVIDIA驱动,选择N。在提示是否安装opengl时,一定要选择N。后面的其他提示都选择默认或者Y。
然后使用以下命令:
修改系统环境变量,在文件末尾加入以下内容。
3、安装cudnn
cuDNN同样需要我们去NVIDIA的官网下载适合cuda版本的deb文件或tgz文件,我以cudnn-9.1-linux-x64-v7.1.tgz的安装为例。进入压缩包所在的目录分别执行以下命令:
4、anaconda的安装
到官网下载安装包,我使用的是 Python 3.5 版本的Anaconda3-5.2.0-Linux-x86_64.sh。在终端进入安装包所在目录,执行命令:
在安装过程中在选择加入环境变量时,选择yes。
创建虚拟环境,以备后续安装tensorflow。
二、安装tensorflow
下面介绍通过Bazel编译tensorflow源码的方式在虚拟环境安装tensorflow。
5、安装Bazel
到github下载bazel-0.14.1-installer-linux-x86_64.sh
进入安装包目录,执行命令:
将export PATH="$PATH:$HOME/bin"添加到~/.bashrc中。
6、编译安装tensorflow
到github下载源码,branch处可以选择版本为r1.8。解压源码,从终端进入解压文件目录,执行以下命令:
按照提示输入y/n或者路径等信息。安装过程中cuda选择9.1版本,cudnn选择7.1版本。
配置完成后依次执行以下命令,安装tensorflow
7、运行例程
我按照以上步骤安装好TensorFlow之后,运行例程,出现了以下问题:
这个问题与编译文件无关,其原因在于安装cuda9.1的时候有一些配置文件没有正确进行配置,也就是一些文件找不到。找不到并不是意味着不存在,而是没有通过正确的路径来查找。依次执行以下命令(软连接):
将相应的文件和你的cuda路径进行软连接,这是默认安装路径,如果你的路径不一样,需要修改上面的代码!软连接后就可以正常import TensorFlow了。cuda,cudnn正常工作。
参考链接:
https://cloud.tencent.com/developer/article/1150020
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/
以前已经发过在
win10上安装cpu版本tensorflow
的教程。有需求的同学可以出戳链接看一看。
领取专属 10元无门槛券
私享最新 技术干货