tensorflow、keras安装以及遇到的问题

tensorflow、keras安装以及遇到的问题

操作系统

ubuntu 16.04及以上

安装版本说明

为了避免tensorflow、keras、cuda、cudnn出现版本不兼容问题，分别选择以下版本：
tensorflow: tensorflow-gpu 1.4
keras: keras 2.0.8
cuda: cuda 8.0
cudnn: cudnn 6.0
注意：tensorflow的gpu版本目前只支持cuda 8.0, 而不支持cuda 9.0以上，这是在安装过程中大概率碰到的问题。

安装过程

安装tensorflow

pip install tensorflow-gpu==1.4

安装keras

pip install keras==2.0.8

安装cuda：
从官网下载https://developer.nvidia.com/cuda-80-ga2-download-archive相关安装包以及补丁，并按以下方式安装

dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
apt-get update
apt-get install cuda

安装cudnn:
从https://developer.nvidia.com/rdp/cudnn-download 下载相关安装文件， 并按以下方式安装

cp cuda/include/cudnn.h /usr/local/cuda