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

没有报错,说明全部安装成功!!!

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值