Ubuntu-install-keras-cuda-tensorflow/theano

参考原网址：https://github.com/BIGBALLON/cifar-10-cnn/blob/master/doc/Ubuntu-install-cuda-tensorflow.md

特别声明：原网址上的大牛已经写的很详细了，我只是根据自己遇到的问题，做出了部分修改。

STEP 1: Install Ubuntu 16.04

到 Ubuntu官网 下载ISO镜像, 刻录进U盘
进入Win10，打开磁盘管理，压缩出足够的的磁盘空间(40GB or more)
Reboot，进入BIOS，关闭 Security boot 及 Win10 的 Fast boot (Important)
Reboot，从USB 引导进入安装界面
选择Ubuntu与Windows共存，一路安装到底

STEP 2: Install NVIDIA Driver

更新源和必要的软件，如果在国内请自行更换合适的source

sudo apt-get update
sudo apt-get upgrade

禁用Nouveau

sudo vi /etc/modprobe.d/disable-nouveau.conf

//加入如下两行

blacklist nouveau
options nouveau modeset=0

重建kernel initramfs并重新启动

sudo update-initramfs -u
sudo reboot

安装NVIDIA 驱动

重启进入登录界面，切换到tty1(ctrl+alt+f1), 关闭lightdm图形界面

sudo service lightdm stop

增加 Nvidia 的 ppa 源

sudo apt-get purge nvidia-*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

安装Nvidia

sudo apt-get install nvidia-375
reboot

重启，再次进入tty1，执行如下命令，没问题则ok了

sudo apt-get update && sudo apt-get upgrade

最后，用 nvidia-smi 查看GPU的信息

nvidia-smi

STEP 3: Install CUDA 8.0

从官网下载CUDA文件(以cuda_8.0.61_375.26_linux.run为例)
加执行权限并安装
安装时会询问是否安装显卡驱动，务必选择No，前面已自行安装

cd Downloads/
sudo chmod a+x cuda_8.0.61_375.26_linux.run
sudo ./cuda_8.0.61_375.26_linux.run

配置环境变量,直接放在系统配置文件 profile 里面

gedit /etc/profile
//加入这两行， LD_LIBRARY_PATH 和 CUDA_HOME 都不能少
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

保存后退出

测试CUDA的Sample

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery

STEP 4: Install Cudnn

到 官网 下载(需注册账号)， 解压 & 复制文件 & 加执行权限

cd Downloads/
sudo tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/include/
cd cuda/lib64/
sudo cp lib* /usr/local/lib/
cd /usr/local/lib
ls
下面根据实际情况修改
sudo chmod +r libcudnn.so.6.0.21
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so
sudo ldconfig

STEP 5: Install Tensorflow via pip

安装必要依赖

sudo apt-get install libcupti-dev

安装python-pip python-dev 并更新到最新版

sudo apt-get install python-pip python-dev
pip install -U pip
//or python3
sudo apt-get install python3-pip
pip3 install --upgrade pip

安装tensorflow

//python
sudo pip install tensorflow-gpu
//or python3
sudo pip3 install tensorflow-gpu

STEP 6: Install theano via pip

//python
sudo pip install theano
//or python3
sudo pip3 install theano

STEP 7: Install keras via pip

安装keras

//python
sudo pip install keras
//or python3
sudo pip3 install keras

测试

开启一个terminal,这里我测试py3下的tensorflow。

bg@cgilab:~$ python3

>>> import keras
>>> import tensorflow as tf
>>> import theano

没有报错，说明全部安装成功！！！