Ubuntu14.04+GTX1070 配置tensorflow

前端之家收集整理的这篇文章主要介绍了Ubuntu14.04+GTX1070 配置tensorflow前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

一、 准备工作

1. 首先下载英伟达驱动,下载地址。根据自己的系统以及显卡型号选择驱动,笔者下载的是NVIDIA-Linux-x86_64-367.57.run

2. 下载CUDA,下载地址。笔者选择的是CUDA8.0,目前1070和1080好像只支持8.0。其他显卡的也可以选择CUDA7.5.

3. 下载cudnn,下载地址。要填写一个小问卷,很快的。根据自己选择的CUDA版本选择cudnn版本,笔者选择的是cudnn v5.1 for CUDA8.0

4. 下载tensorflow,下载地址。这里我们选择安装编译好的二进制包,这是最简单的方法。在下载页面如下图位置,选择Linux-GPU:Python3.5


二、 安装显卡驱动、CUDA、cudnn

1. 安装显卡驱动

如果是新安装的Ubuntu系统,那么直接安装按以下步骤安装即可。如果曾经装过显卡驱动,请先卸载旧版驱动。

a) Ctrl+Alt+F1进tty,如果黑屏可以修改grub文件。具体做做法如下

sudo gedit /etc/default/grub 用gedit打开grub文件

将GRUB_CMDLINE_LINUX_DEFAULT的值修改为 quiet splash nomodeset

sudo update-grub 更新grub

重启电脑(可能会发现系统分辨率变了,不要紧,安装好驱动就会恢复正常)

b) 进入tty之后,输入用户名和密码。

c) sudo service lightdm stop 关闭图形界面

cd DL 切换到你放驱动的文件夹,笔者的在DL文件

sudo chmod 755 NVIDIA-Linux-x86_64-367.57.run获取权限,根据自己驱动名修改

sudo ./NVIDIA-Linux-x86_64-367.57.run 如果提示pre-install失败可以忽略

sudo service lightdm start 开启图形界面

d)测试一下,Ctrl+Alt+t打开terminal

cat /proc/driver/nvidia/version

如果出现下面的界面

那么说明驱动安装成功了,可以看到笔者安装的版本是367.57

2. 安装CUDA

a) 首先安装相关的依赖库

sudo apt-get install freeglut3-dev
sudo apt-get install build-essential
sudo apt-get install libx11-dev
sudo apt-get install libxmu-dev
sudo apt-get install libxi-dev
sudo apt-get install libglu1-mesa
sudo apt-get install libglu1-mesa-dev

b) 切换到放cuda安装包的文件

sudo sh cuda_8.0.44_linux.run --override

c) 这时候会出现一大堆协议,一直按space就可以,最后accept

d) 需要特别注意:安装过程中问你要不要Install NVIDIA ~~ drive一定要选n,不然之前装的显卡驱动就白装了。这里主要是因为cuda提示安装的驱动版本比较低。

e) 其他都选yes,默认文件夹不用去改它。

f) 安装完毕之后设置一下环境变量。

将下面两行写到~/.bashrc文件中(sudo gedit ~/.bashrc)

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda8.0-/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

g) 测试一下,切换到cuda_sample文件

cd ~/NVIDIA_CUDA-8.0_Samples 切换到sample文件
make make一下
cd bin/x86_64/linux/release 切换到release目录
./deviceQuery 运行一下实例

如果出现下面的图,说明安装成功了。

3. 安装cudnn

a) cd DL 切换目录,根据实际情况更改
tar xvzf cudnn-8.0-linux-x64-v5.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*

这样子就可以了。

4. 安装anacond

anaconda集成了python的100多个库,比较方便。

先去官网下载,下载地址

下载好后,切换到下载目录

bash Anaconda3-4.2.0-Linux-x86_64.sh

中间会问你要不要修改环境变量,选y

安装完成记得执行一下 source ~/.bashrc 才会生效

测试一下,在命令行输入python,出现下图说明安装正确。

5. 安装tensorflow

a) 首先执行下面两句,开启GPU加速

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

b) pip install tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl

这就安装好了。你没看错。如果是自己折腾从源码编译不要太麻烦

c) 测试一下

cd anaconda3/lib/python3.5/site-packages/tensorflow/models/image/mnist

python convolutional.py

5.4ms,比cpu不要快太多。就这么愉快的结束了。真正的学习过程才刚刚开始~

猜你在找的Ubuntu相关文章