Spaces:
Building
on
A10G
Building
on
A10G
File size: 1,272 Bytes
58508da eae186f 58508da 466db9e 58508da 04ac67a 4a32a7f 466db9e 4a32a7f 2a0c586 11f5e70 466db9e 1cff57f 1d90460 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
FROM nvcr.io/nvidia/pytorch:24.01-py3
RUN apt-get update && \
apt-get install -y \
wget \
git \
gnutls-bin \
openssh-client \
libghc-x11-dev \
gcc-multilib \
g++-multilib \
libglew-dev \
libosmesa6-dev \
libgl1-mesa-glx \
libglfw3 \
xvfb \
mesa-utils \
libegl1-mesa \
libgl1-mesa-dev \
libglu1-mesa-dev \
libglib2.0-0 \
libsm6 \
libxrender1 \
libxext6 \
unzip \
openjdk-8-jdk
RUN useradd -m -u 1000 user
USER user
ENV PATH="/home/user/.local/bin:$PATH"
WORKDIR $HOME/app
RUN git clone https://github.com/CraftJarvis/ROCKET-1.git && \
cd ROCKET-1 && \
git pull && \
python -m pip install --upgrade pip && \
pip install -e . && \
cd rocket/realtime_sam && \
pip install -e . && \
cd checkpoints && \
bash download_ckpts.sh
WORKDIR $HOME/app/ROCKET-1/rocket/stark_tech
RUN python -c "from huggingface_hub import hf_hub_download;hf_hub_download(repo_id='phython96/ROCKET-MCP-Reborn', filename='MCP-Reborn.zip', local_dir='.')" && \
unzip -q MCP-Reborn.zip && rm MCP-Reborn.zip
WORKDIR $HOME/app/ROCKET-1/rocket/arm
RUN git pull
CMD ["python", "eval_rocket.py", "--port", "7860", "--sam-path", "/app/ROCKET-1/rocket/realtime_sam/checkpoints"]
|