Ubuntu16+GTX1050Ti配置tensorflow

本文主要参考了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
[install]
trusted-host=mirrors.aliyun.com

$ 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:"

Ubuntu16+GTX1050Ti配置tensorflow

猜你喜欢