平台:ubuntu 16.04 显卡:NVIDIA TITAN XP
此处是以tensorflow作为后端
一、安装TensorFlow GPU版
前面写过教程,参考:https://blog.csdn.net/u014797226/article/details/80229887
二、安装python相关包
(1)安装BLAS库
sudo apt-get install build-essential cmake git unzip \pkg-config libopenblas-dev liblapack-dev
(2) 安装python科学套件:Numpy、Scipy和Matplotlib
sudo apt-get install python-numpy
sudo apt-get install python-scipy
sudo apt-get install python-matplotlib
sudo apt-get install python-yaml
(3) 安装HDF5 (用二进制格式保存大文件)
sudo apt-get install libhdf5-serial-dev python-h5py
(4) 安装可视化工具
sudo apt-get install graphviz
sudo pip install pydot-ng
(5) 安装画图工具
pip3 install matplotlib --user
(6) 安装opencv
sudo pip3 install opencv-python
注:此处一定要用pip3安装,不然import cv2的时候会出现错误。
三、安装keras
sudo pip install keras
注:由于网络原因中间失败过几次,重新安装即可
四、测试
在python环境下输入import keras,没有问题则说明安装成功
五、运行mnist示例
git clone https://github.com/fchollet/keras
cd keras-master/examples
python mnist_cnn.py