docker部署深度学习模型

本文详述了如何在Ubuntu18.04上使用Docker部署基于Python3.6.8、Flask1.1.1、Keras2.2.4和TensorFlow1.14CPU版的深度学习图像分类模型。首先,介绍了Docker的基本操作,包括获取镜像、运行容器和进入容器。接着,详细阐述了如何在容器内安装Python及相关库,如设置国内源、安装vim、sudo、wget、zlib、python3.6.8、pip、setuptools等。然后,部署模型并确保其正常运行。最后,讲解了如何保存容器为镜像,以便于未来快速部署到新的服务器。
摘要由CSDN通过智能技术生成

准备工作

本文目标为在docker上部署深度学习模型,并将完成的实例保存为镜像,方便今后部署。本次部署用到了docker、ubuntu18.04、python3.6.8、flask1.1.1、keras2.2.4、numpy1.17、tensorflow1.14cpu版(因为模型为图像分类模型、推理过程不需要太大的算力,在服务器至强6278C CPU上一张图的响应时间在0.4秒以内,相比之下,在本地服务器gtx 2080ti GPU上用时为0.24秒左右,并没有太大提升)以及常用的python库。

docker基本操作

1.docker镜像的获得

docker search ubuntu:18.04

镜像搜索结果

选择一个镜像,通过该路径下载镜像,如:

docker pull docker.io/library/ubuntu:18.04

该路径为官方镜像,里面很多东西都莫得,甚至没有python3,都需要自己装,如果想要省事就装一个配置完整的。
2.通过镜像运行一个容器

docker run -dit -v /root/mzw:/root/mzw -p 5001:5000 --name=test9999 docker.io/ubuntu

其中-dit,-v,-p,–name都是参数,详细参数介绍见

  • 1
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值