本文详细介绍如何安装NVIDIA Docker并验证GPU支持，随后指导如何选择及下载Keras-GPU特定版本，包括运行环境配置，适用于深度学习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
扫一扫
1246
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
