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

# 如果只是临时使用 也可以
# $ 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:"
 

 

你可能感兴趣的:(Ubuntu16+GTX1050Ti配置tensorflow)