利用docker安装keras镜像

本身并不复杂的操作,由于自己实在太小白,折腾了好久,现在还是有不明白的地方,不过先安装上再慢慢学习。

1、制作keras的docker镜像
GitHub上keras的源码是提供了Dockerfile的,可以用来制作镜像,网址:
https://github.com/keras-team/keras/blob/master/docker/Dockerfile
但是由于官网提供的是GPU版本,而我们用的虚拟机,需要安装CPU版本的,因此对Dockerfile做修改,修改后(由于粘贴整个文件内容用markdown会出奇怪的问题,所以这里直接上传文件):
修改后的Dockerfile文件

build命令生成镜像:

docker build -t keras_docker /root/keras_dockerfile

keras_docker 是自己定义的生成镜像的标签,后面是修改后dockerfile所存放的目录

2、run:

docker run -it -p 8886:8888 -v /keras:/notebooks/kerasdata keras_docker:latest

冒号前面是上一步build时所定义的标签

3、在浏览器访问 服务器地址:8886
输入token后的口令
注意这里和安装tensorflow镜像一样,也是不要停止运行程序,可以关掉这个连接让它在后台运行

4、测试
运行example的方法,进入example文件夹,新建一个python文档(ipynb文件),然后在cell内输入%run mnist_cnn.py,并运行,即可成功运行example文件夹中的mnist_cnn.py文件

由于运行时需要下载数据集,可能出现网络不稳定而报timeouterror的情况,再次尝试即可成功运行。

猜你喜欢

转载自blog.csdn.net/buki26/article/details/80612699