Spaces:
Paused
Paused
FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04 | |
WORKDIR /content | |
ENV PATH="/home/camenduru/.local/bin:${PATH}" | |
RUN adduser --disabled-password --gecos '' camenduru && \ | |
adduser camenduru sudo && \ | |
echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers && \ | |
chown -R camenduru:camenduru /content && \ | |
chmod -R 777 /content && \ | |
chown -R camenduru:camenduru /home && \ | |
chmod -R 777 /home | |
RUN apt update -y && add-apt-repository -y ppa:git-core/ppa && apt update -y && apt install -y aria2 git git-lfs unzip ffmpeg | |
USER camenduru | |
RUN pip install -q opencv-python imageio imageio-ffmpeg ffmpeg-python av xformers==0.0.25 runpod \ | |
pytorch-lightning==2.1.2 gradio==3.50.2 einops omegaconf torchmetrics webdataset accelerate tensorboard \ | |
PyMCubes trimesh rembg transformers diffusers==0.20.2 bitsandbytes imageio[ffmpeg] xatlas plyfile \ | |
git+https://github.com/NVlabs/nvdiffrast jax==0.4.19 jaxlib==0.4.19 ninja | |
RUN GIT_LFS_SKIP_SMUDGE=1 git clone https://github.com/TencentARC/InstantMesh /content/InstantMesh | |
COPY ./worker_runpod.py /content/InstantMesh/worker_runpod.py | |
WORKDIR /content/InstantMesh | |
CMD python worker_runpod.py |