参考原网址: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
没有报错,说明全部安装成功!!!