安装TensorFlow-

安装tensorflow-gpu:

1、安装cuda-repo-9.0.deb

sudo dpkg -i cuda*
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
sudo apt-get update
sudo apt-get install cuda

2、

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
source ~/.bashrc

3、安装cudnn7.0

tar -zxvf cudnn-7.0*.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp cuda/lib64/* /usr/local/cuda-9.0/lib64/
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*

安装h5py:

(pip3 install cython)
sudo apt-get install libhdf5-dev
pip3 install h5py