Docker-keras-tf-gpu
安装nvidia-dockersudo docker run --runtime=nvidia --rm nvidia/cuda:9.0-base查看是否安装成功sudo nvidia-docker run --rm nvidia/cuda:9.0-base nvidia-smi安装keras-gpuhttps://store.docker.com/community/imag...
·
安装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
权威｜前沿｜技术｜干货｜国内首个API全生命周期开发者社区
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)