Tensorflow-gpu、keras安装

1、python3、pip3安装

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install python3-pip python3-dev

2、安装Python科学套件

BLAS库

sudo apt-get install build-essential cmake git unzip pkg-config libopenblas-dev liblapack-dev

Numpy、SciPy、Matplotlib

sudo apt-get install python3-numpy python3-scipy python3-matplotlib python3-yaml

HDF5，用于保存数值数据

sudo apt-get install libhdf5-serial-dev python3-h5py

安装Graphviz与pydot-ng用于kears 模型可视化

sudo apt-get install graphviz
sudo pip3 install pydot-ng

opencv

sudo apt-get install python3-opencv

3、安装tensorflow-gpu

sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ tensorflow-gpu

上面为指定清华pip源，下载速度更快；tensorflow-gpu可指定安装版本，如指定1.12版本：tensorflow-gpu==1.12

4、安装kears

sudo pip3 install keras