Docker-keras-tf-gpu

安装nvidia-docker

sudo docker run --runtime=nvidia --rm nvidia/cuda:9.0-base

查看是否安装成功

sudo nvidia-docker run --rm nvidia/cuda:9.0-base nvidia-smi

安装keras-gpu

https://store.docker.com/community/images/gw000/keras/tags

选择对应版本，这里选择2.1.4-py3-tf-gpu，注意下载要写全gw000/keras:2.1.4-py3-tf-gpu

下载命令

sudo docker pull gw000/keras:2.1.4-py3-tf-gpu

运行

sudo docker run -it --runtime=nvidia gw000/keras:2.1.4-py3-tf-gpu bash

sudo nvidia-docker run -it gw000/keras:2.1.4-py3-tf-gpu