基于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)
相关推荐
晓风残月淡1 分钟前
JVM字节码与类的加载(一):类的加载过程详解
开发语言·jvm·python
Akshsjsjenjd28 分钟前
docker网络
网络·docker·容器
ygyqinghuan30 分钟前
PyTorch 实现 MNIST 手写数字识别
人工智能·pytorch·python
reasonsummer33 分钟前
【办公类-115-05】20250920职称资料上传04——PDF和PDF合并PDF、图片和PDF合并PDF(十三五PDF+十四五图片)
java·python·pdf
HsuHeinrich40 分钟前
利用径向柱图探索西班牙语学习数据
python·数据可视化
独行soc1 小时前
2025年渗透测试面试题总结-105(题目+回答)
网络·python·安全·web安全·adb·渗透测试·安全狮
史锦彪1 小时前
用 PyTorch 实现 MNIST 手写数字识别:从入门到实践
人工智能·pytorch·python
董建光d1 小时前
PyTorch 实现 MNIST 手写数字识别完整流程(含数据处理、模型构建与训练可视化)
人工智能·pytorch·python
小宁爱Python2 小时前
从零搭建 RAG 智能问答系统3:聊天信息持久化和登录注册
python
天才少女爱迪生3 小时前
LLVM(Low Level Virtual Machine)介绍
python·数据挖掘