content: FROM nvcr.io/nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
  
  ENV DEBIAN_FRONTEND noninteractive
  
  # Install package dependencies
  RUN apt-get update && \
      apt-get install -y --no-install-recommends \
          build-essential \
          autoconf \
          automake \
          libtool \
          pkg-config \
          ca-certificates \
          wget \
          git \
          curl \
          libjpeg-dev \
          libpng-dev \
          language-pack-en \
          locales \
          locales-all \
          python3 \
          python3-dev \
          python3-pip \
          python3-setuptools \
          libprotobuf-dev \
          protobuf-compiler \
          zlib1g-dev \
          swig \
          vim \
          gdb \
          valgrind \
          libsm6 \
          libxext6 \
          libxrender-dev && \
      apt-get clean
  
  RUN cd /usr/local/bin && \
      ln -s /usr/bin/python3 python && \
      ln -s /usr/bin/pip3 pip && \
      pip install --upgrade pip setuptools wheel
  
  # System locale
  # Important for UTF-8
  ENV LC_ALL en_US.UTF-8
  ENV LANG en_US.UTF-8
  ENV LANGUAGE en_US.UTF-8
  
  RUN pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0
  RUN pip install scikit-learn==1.1.3
  