sam2 docker部署

目录

DockerFIle

torchserve启动报错：

缺少nvgpu，

---

DockerFIle

# 第一阶段：构建环境
FROM opedromartins/sam2-docker:sam2.1# 设置清华源（系统级别）
RUN sed -i 's/archive.ubuntu.com/mirrors.tuna.tsinghua.edu.cn/g' /etc/apt/sources.list && \sed -i 's/security.ubuntu.com/mirrors.tuna.tsinghua.edu.cn/g' /etc/apt/sources.list# 设置工作目录
WORKDIR /app# 设置pip清华源并安装所有依赖
RUN pip install --no-cache-dir --no-warn-script-location \torchserve torch-model-archiver nvgpu

docker build -t torchserve-seg:sam .

torchserve启动报错：

torchserve --start --model-store /app/model-store --models sam2-segmenter=sam2-segmenter.mar --ncs --disable-token java not found, please make sure JAVA_HOME is set properly.

解决方法，你用的是 Ubuntu 镜像，可以加上：

RUN apt-get update && apt-get install -y openjdk-11-jdk

缺少nvgpu，

安装，必须带上清华源。

pip install nvgpu -i https://pypi.tuna.tsinghua.edu.cn/simple