安裝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

選擇對應版本，這裡選擇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