pytorch3d GPU Dockerfile for CUDA 11.7

Dockerfile

FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04

ARG DEBIAN_FRONTEND=noninteractive
ENV TZ=Europe/Berlin

RUN apt-get update \
    && apt-get install -y software-properties-common \
    && add-apt-repository -y ppa:deadsnakes/ppa \
    && apt-get update \
    && apt-get install -y python3.9 python3.9-distutils python3.9-dev python3.9-venv \
    && apt-get upgrade -y\
    && apt-get install -y wget \
    build-essential \
    manpages-dev \
    ca-certificates \
    cmake \
    dos2unix \
    libglib2.0-0 \
    nano \
    ninja-build \
    g++ \
    git \
    libglfw3-dev \
    libgles2-mesa-dev \
    && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 \
    && apt-get install -y python3-pip \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/* \
    && gcc --version \
    && pip install -U fvcore iopath scikit-image matplotlib imageio plotly opencv-python \
    && pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

RUN nvcc --version \
    && FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX" CUDA_HOME='/usr/local/cuda' pip install git+https://github.com/facebookresearch/pytorch3d.git@stable \
    && pip install \
      accelerate \
      diffusers \
      chumpy \
      einops \
      face_alignment \
      gradio \
      kornia \
      matplotlib \
      moviepy \
      numpy==1.22.4 \
      numba \
      onnxruntime-gpu \
      opencv-python \
      Pillow \
      psutil \
      scipy \
      scikit-image \
      scikit-video \
      streamlit \
      taming-transformers-rom1504 \
      tqdm \
      transformers \
      watchdog