Tensorflow-GPU版本安装(Ubuntu14.04LTS+Cuda8+Quadro K1200)

前端之家收集整理的这篇文章主要介绍了Tensorflow-GPU版本安装(Ubuntu14.04LTS+Cuda8+Quadro K1200)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

前言

之前装过Caffe,踩过一些坑,现在装Tensorflow相对从容一些,不过还是值得记录一下流程。

我的系统配置是:

系统:Ubuntu 14.04LTS
显卡:Nvidia Quadro K1200

CUDA 8.0安装

1.安装必要的一些库

sudo apt-get install build-essential

  1. 下载CUDA toolkit 8.0 官网地址

友情提示:最好下载本地文件安装.

3. 清楚残留以及禁用nouveau
“`
sudo apt-get –purge remove nvidia-*
vim /etc/modprobe.d/blacklist-nouveau.conf
//写入下面两行
blacklist nouveau
options nouveau modeset=0

//重新生成内核initramfs
sudo update-initramfs -u
//重启
reboot
```

4. 正式安装

//CTRL + ALT +F1 进入tty1模式
sudo service lightdm stop
//更改权限
sudo chmod +x (你的run文件)
sudo sh (你的run文件) --no-opengl-libs//记得别安装opengl库
//在~/.bashrc中添加环境变量
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
//验证驱动版本和CUDA驱动版本
cat /proc/driver/nvidia/version
nvcc -V
sudo service lightdm stop
reboot
  1. 编译CUDA样例并运行
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

安装Nvidia cuDNN

下载好相应版本的cuDNN,我下载的是cuDNN5.1,解压,设置环境变量

sudo tar -xvf (你的cuDNN文件) -C /usr/local
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
//最后记得source ~/.bashrc
cat /proc/driver/nvidia/version //查看驱动版本
nvcc -V //查看CUDA版本
nvidia-smi //查看GPU使用情况

安装Tensorflow-GPU

安装Anaconda3

Anaconda可以省去很多麻烦,建议采用这种方式;
清华大学开源软件镜像站下载想的版本;

//安装Anaconda3
bash Anaconda(你的版本).sh
//环境变量
export PATH=(你的安装路径):$PATH //比如 /root/Anaconda3/bin:$PATH
conda list //查看已经安装的库
conda update pip //更新pip

finally,最后一步!

..................................
//第一种安装方式
//pip 安装,网速不稳定
pip install tensorflow-gpu
...................................
//第二种安装方式
//从清华开源软件镜像站下载需要的版本,建议1.2,1.3报错(cuDNN版本太低),网址https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux
pip install tensorflow_gpu-1.2.0-cp36-cp36m-linux_x86_64.whl

哈哈,最后测试啦!

python
import tensorflow as tf

搞定!

注意:
import没错并不意味着能够用GPU进行训练,需要用下面的命令进行检测,不然就重装吧,重装一次的飘过,囧-_-

猜你在找的Ubuntu相关文章