当前位置：首页 > news >正文

gradio-osprey-demo

news 2025/8/27 5:43:00

创建需要的dockerfle

###################

# 使用 Ubuntu 作为基础镜像
FROM nvcr.io/nvidia/cuda:11.8.0-devel-ubuntu22.04

# 更新软件包列表并安装依赖项
RUN apt update && \
apt install -y python3 python3-pip git ffmpeg libsm6 libxext6 curl wget vim libgl1 && \
cp /usr/bin/python3 /usr/bin/python && \
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \
pip install git+https://github.com/facebookresearch/segment-anything.git && \
pip install opencv-python pycocotools matplotlib onnxruntime onnx

# 设置 Node.js 环境变量
ENV NODE_OPTIONS="--max-old-space-size=8192"

# 安装 nvm、Node.js 和 pnpm

ENV NVM_DIR /root/.nvm
RUN mkdir -p $NVM_DIR && curl https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.3/install.sh | bash && . ~/.bashrc && nvm install v18.16.0
RUN . $NVM_DIR/nvm.sh && ln -s $NVM_DIR/$(nvm current) $NVM_DIR/cur && ln -s $NVM_DIR/versions/node/$(nvm current) $NVM_DIR/versions/node/cur
ENV NODE_PATH $NVM_DIR/cur/lib/node_modules
ENV PATH $NVM_DIR/versions/node/cur/bin:$PATH

ADD pnpm /usr/local/bin/
RUN chmod a+x /usr/local/bin/pnpm

# 设置环境变量
ENV NODE_OPTIONS=--max_old_space_size=6000
# 克隆并安装 Gradio 应用
RUN git clone https://github.com/LiWentomng/gradio-osprey-demo.git && \
cd gradio-osprey-demo && \
bash scripts/build_frontend.sh && \
pip install -e .

# 设置工作目录
WORKDIR /gradio-osprey-demo

# 可选：暴露端口
# EXPOSE <port>

# 可选：指定容器启动时的命令
CMD sleep 10000000000

查看全文

http://www.lryc.cn/news/271705.html