基于Dockerfile打包算法镜像

java 复制代码
# 使用 NVIDIA CUDA 12.0.1 运行时的 Ubuntu 20.04 基础镜像
FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04

# 更新包列表并安装必要的包
RUN apt-get update && apt-get install -y --no-install-recommends \
    python3 \
    python3-pip \
    python3-venv \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

# 创建并激活虚拟环境
RUN python3 -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"

# 安装 PyTorch
RUN pip install torch torchvision torchaudio

# 安装 FastAPI
RUN pip install fastapi uvicorn

# 将代码复制到容器中
COPY . /HotSpots

#工作目录
WORKDIR  /HotSpots

# 暴露端口
EXPOSE 8000

# 运行 FastAPI 应用
CMD ["sh","start.sh"]

start.sh 文件

bash 复制代码
python infer.py

infer.py文件

bash 复制代码
if __name__ == '__main__':
    uvicorn.run("infer:app",host="0.0.0.0",port=8000)
相关推荐
CodeCraft Studio31 分钟前
CAD文件处理控件Aspose.CAD教程:使用 Python 将绘图转换为 Photoshop
python·photoshop·cad·aspose·aspose.cad
Python×CATIA工业智造2 小时前
Frida RPC高级应用:动态模拟执行Android so文件实战指南
开发语言·python·pycharm
onceco3 小时前
领域LLM九讲——第5讲 为什么选择OpenManus而不是QwenAgent(附LLM免费api邀请码)
人工智能·python·深度学习·语言模型·自然语言处理·自动化
狐凄4 小时前
Python实例题:基于 Python 的简单聊天机器人
开发语言·python
悦悦子a啊5 小时前
Python之--基本知识
开发语言·前端·python
笑稀了的野生俊6 小时前
在服务器中下载 HuggingFace 模型:终极指南
linux·服务器·python·bash·gpu算力
Naiva6 小时前
【小技巧】Python+PyCharm IDE 配置解释器出错,环境配置不完整或不兼容。(小智AI、MCP、聚合数据、实时新闻查询、NBA赛事查询)
ide·python·pycharm
cui_hao_nan6 小时前
Docker后端部署
运维·docker·容器
路来了7 小时前
Python小工具之PDF合并
开发语言·windows·python
蓝婷儿7 小时前
Python 机器学习核心入门与实战进阶 Day 3 - 决策树 & 随机森林模型实战
人工智能·python·机器学习