gradio-osprey-demo

创建需要的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

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