diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile index 0e5268d311..0e56f2fa93 100644 --- a/dockers/base-cuda/Dockerfile +++ b/dockers/base-cuda/Dockerfile @@ -59,7 +59,6 @@ RUN \ add-apt-repository ppa:deadsnakes/ppa && \ apt-get install -y \ python${PYTHON_VERSION} \ - python3-setuptools \ python${PYTHON_VERSION}-dev \ && \ update-alternatives --install /usr/bin/python${PYTHON_VERSION%%.*} python${PYTHON_VERSION%%.*} /usr/bin/python${PYTHON_VERSION} 1 && \ @@ -79,6 +78,8 @@ RUN \ curl https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} && \ # Disable cache \ pip config set global.cache-dir false && \ + # Install recent setuptools to obtain pkg_resources \ + pip install setuptools==75.6.0 && \ # set particular PyTorch version \ pip install -q wget packaging && \ python -m wget https://raw.githubusercontent.com/Lightning-AI/utilities/main/scripts/adjust-torch-versions.py && \