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