dockerfile

FROM huggingface/transformers-pytorch-gpu

RUN apt-get update
RUN apt-get -y install libglfw3-dev libgles2-mesa-dev libgl1 freeglut3-dev unzip ffmpeg libsm6  libxext6 libgl1-mesa-dri libegl1-mesa libgbm1 build-essential
RUN python3 -m pip install --upgrade pip
RUN pip3 install opencv-python numpy trimesh plyfile pyyaml scikit-image scikit-learn chumpy tensorboard kornia loguru pycocotools yacs lmdb fire setuptools einops tqdm ipython gym transformers==4.44.2 OmegaConf matplotlib diffusers rembg segment_anything jaxtyping imageio iopath timm open3d pyrender pytorch-lightning smplx==0.1.28 chumpy @ git+https://github.com/zerchen/chumpy xtcocotools pandas hydra-core hydra-submitit-launcher hydra-colorlog pyrootutils rich webdataset ultralytics dill git+https://github.com/facebookresearch/pytorch3d.git@stable git+https://github.com/zerchen/lang-segment-anything.git typeguard gradio==5.0.2

RUN wget -q https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run -o /tmp/cuda121.run
RUN chmod +x /tmp/cuda121.run && /tmp/cuda121.run --silent --toolkit
RUN cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app

ENTRYPOINT ["python3"]
CMD ["app.py"]