本文主要参考了http://blog.csdn.net/sikao_luwei/article/details/69375126的方法,并根据自己实际情况加以精简完善!
官方安装说明文档:https://www.tensorflow.org/install/install_linux
第一部分:显卡驱动安装
首先需要卸载旧版驱动 sudo apt-get autoremove --purge nvidia-*
step1: CUDA和nvidia驱动编译安装需要内核头文件
$ sudo apt-get install linux-headers-$(uname -r)
step2: 安装nvidia驱动必须禁用nouveau,所以创建一个文件
$ cd /etc/modprobe.d/
$ sudo vim blacklist-nouveau.conf
在文件中写入以下内容
blacklist nouveau
options nouveau modeset=0
重新生成内核initramfs:
$ sudo update-initramfs -u
为保证禁用成功,重启系统,执行 >>lsmod | grep nouveau 确保没有输出即可
step3: 按Ctrl + Alt + F1进入图像界面,关闭X servers
$ sudo service lightdm stop
step4: 安装NVIDIA驱动 驱动地址http://www.geforce.cn/drivers
$ sudo chmod a+x NVIDIA-Linux-x86_64-384.98.run
$ sudo ./NVIDIA-Linux-x86_64-384.98.run --no-opengl-files
报错The distribution-provided pre-install script failed!不必理会,继续安装。
重要!!!遇到"Would you like to run the nvidia-xconfig utility to automatically update your X configuration file...."一定要选yes
step5:重启X-window服务
sudo service lightdm start
step6: 查看驱动是否安装成功
$ nvidia-smi
也可以采用另一种方法(推荐! 而且建议不使用cuda中的驱动,最好驱动和cuda分开安装)
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get update
$ sudo service lightdm stop
$ sudo apt-get install nvidia-384
第二部分:CUDA和CUDnn安装
step1: 安装CUDAToolkit 8.0
cuda下载地址:https://developer.nvidia.com/cuda-downloads
$ sudo chmod a+x cuda_8.0.61_375.26_linux.run
$ sudo ./cuda_8.0.61_375.26_linux.run
安装遇到Install NVIDIA Accelerated Graphics Driver...??选择no!!!
step2: 设置环境变量
$ vim ~/.bashrc
添加以下内容:
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
$ source ~/.bashrc
step3: 测试CUDA Toolkit是否安装正确
$ nvcc --version
step4: 安装cuDNN v6.0
cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download
注意:cuda和cudnn版本需要匹配
$ tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz
$ sudo cp cuda/include/* /usr/local/cuda/include/
$ sudo cp -d cuda/lib64/* /usr/local/cuda/lib64
第三部分:TensorFlow安装
step1:下载安装Anaconda
下载地址:https://www.anaconda.com/download/#linux
$ sh Anaconda2-5.0.1-Linux-x86_64.sh
也可以直接安装python:
$ sudo apt-get install python-pip python-dev
step2:pip安装TensorFlow
对pip进行换源:
$ sudo vim ~/.pip/pip.conf (如果路径不存在就自己创建)
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
# 如果只是临时使用 也可以
# $ pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
$ pip install tensorflow # Python 2.7; CPU support (no GPU support)
$ pip3 install tensorflow # Python 3.n; CPU support (no GPU support)
$ pip install tensorflow-gpu # Python 2.7; GPU support
$ pip3 install tensorflow-gpu # Python 3.n; GPU support
本人在spyder中运行程序时,遇到 import tensorflow时ImportError: libcudnn.so.6XXXX的问题。解决方法:首先确定/usr/local/cuda/lib64下是否有这个libcudnn文件,如果存在该文件但依旧报错,则执行
$ sudo apt-get install libcupti-dev
"The libcupti-dev library, which is the NVIDIA CUDA Profile Tools Interface. This library provides advanced profiling support. To install this library, issue the following command:"