Dockerfile inf2 -> use ubuntu 22.04

20.04 is reaching end of support this year anyway

remove some unused dependencies for the inference web service to run

Signed-off-by: Raphael Glon <[email protected]>

Author: oOraph

dockerfiles/pytorch/Dockerfile.inf2

@@ -1,22 +1,16 @@
 # Build based on https://github.com/aws/deep-learning-containers/blob/master/huggingface/pytorch/inference/docker/2.1/py3/sdk2.18.0/Dockerfile.neuronx
-FROM ubuntu:20.04 as base
+FROM ubuntu:22.04 as base
 
 LABEL maintainer="Hugging Face"
 
 ARG PYTHON=python3.10
 ARG PYTHON_VERSION=3.10.12
 ARG MAMBA_VERSION=23.1.0-4
 
-# Neuron SDK components version numbers
-# ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.1.0
-# ARG NEURONX_DISTRIBUTED_VERSION=0.7.0
-# ARG NEURONX_CC_VERSION=2.13.66.0
-ARG NEURONX_TRANSFORMERS_VERSION=0.12.313
 ARG NEURONX_COLLECTIVES_LIB_VERSION=2.22.33.0-d2128d1aa
 ARG NEURONX_RUNTIME_LIB_VERSION=2.22.19.0-5856c0b42
 ARG NEURONX_TOOLS_VERSION=2.19.0.0
 
-
 # HF ARGS
 ARG OPTIMUM_NEURON_VERSION=0.0.28
 
@@ -28,27 +22,18 @@ ENV PATH /opt/conda/bin:/opt/aws/neuron/bin:$PATH
 RUN apt-get update \
  && apt-get upgrade -y \
  && apt-get install -y --no-install-recommends software-properties-common \
- && add-apt-repository ppa:openjdk-r/ppa \
  && apt-get update \
  && apt-get install -y --no-install-recommends \
     build-essential \
     apt-transport-https \
     ca-certificates \
     cmake \
     curl \
-    emacs \
     git \
     jq \
-    libgl1-mesa-glx \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
-    openjdk-11-jdk \
-    vim \
     wget \
     unzip \
     zlib1g-dev \
-    libcap-dev \
     gpg-agent \
  && rm -rf /var/lib/apt/lists/* \
  && rm -rf /tmp/tmp* \
@@ -67,11 +52,6 @@ RUN apt-get update \
  && rm -rf /tmp/tmp* \
  && apt-get clean
 
-# https://github.com/docker-library/openjdk/issues/261 https://github.com/docker-library/openjdk/pull/263/files
-RUN keytool -importkeystore -srckeystore /etc/ssl/certs/java/cacerts -destkeystore /etc/ssl/certs/java/cacerts.jks -deststoretype JKS -srcstorepass changeit -deststorepass changeit -noprompt; \
-    mv /etc/ssl/certs/java/cacerts.jks /etc/ssl/certs/java/cacerts; \
-    /var/lib/dpkg/info/ca-certificates-java.postinst configure;
-
 RUN curl -L -o ~/mambaforge.sh https://github.com/conda-forge/miniforge/releases/download/${MAMBA_VERSION}/Mambaforge-${MAMBA_VERSION}-Linux-x86_64.sh \
  && chmod +x ~/mambaforge.sh \
  && ~/mambaforge.sh -b -p /opt/conda \
@@ -94,7 +74,6 @@ RUN curl -L -o ~/mambaforge.sh https://github.com/conda-forge/miniforge/releases
  && /opt/conda/bin/conda clean -ya
 
 RUN conda install -c conda-forge \
-    scikit-learn \
     h5py \
     requests \
  && conda clean -ya \