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