基于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)
相关推荐
程序员杰哥9 分钟前
Pytest之收集用例规则与运行指定用例
自动化测试·软件测试·python·测试工具·职场和发展·测试用例·pytest
Jyywww12125 分钟前
Python基于实战练习的知识点回顾
开发语言·python
朝朝辞暮i1 小时前
从0开始学python(day2)
python
程序员黄同学1 小时前
Python中的列表推导式、字典推导式和集合推导式的性能和应用场景?
开发语言·python
AI小云1 小时前
【Python高级编程】类和实例化
开发语言·人工智能·python
道之极万物灭1 小时前
Python uv虚拟环境管理工具详解
开发语言·python·uv
高洁012 小时前
【无标题】大模型-模型压缩:量化、剪枝、蒸馏、二值化 (2
人工智能·python·深度学习·神经网络·知识图谱
一晌小贪欢2 小时前
Python爬虫第10课:分布式爬虫架构与Scrapy-Redis
分布式·爬虫·python·网络爬虫·python爬虫·python3
代码AI弗森2 小时前
Python × NumPy」 vs 「JavaScript × TensorFlow.js」生态全景图
javascript·python·numpy
paid槮3 小时前
Shell编程基本介绍
python