From 4e11cdbb60558d84ed6393462a5c1c3bd7de7516 Mon Sep 17 00:00:00 2001 From: Tongze Wang Date: Tue, 10 Sep 2024 22:34:56 -0700 Subject: [PATCH] fix: pip on py 3.12 --- tensorflow.dockerfile | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tensorflow.dockerfile b/tensorflow.dockerfile index 479159d..d167453 100644 --- a/tensorflow.dockerfile +++ b/tensorflow.dockerfile @@ -22,7 +22,7 @@ sudo apt-get update sudo apt install -y software-properties-common sudo add-apt-repository -y ppa:deadsnakes/ppa sudo apt-get update -sudo apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev +sudo apt-get install -y python${PYTHON_VERSION}-full python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-distutils sudo apt-get clean sudo rm -rf /var/lib/apt/lists/* EOF @@ -30,8 +30,9 @@ EOF WORKDIR /home/rocm/tensorflow # TODO: simplify dependencies -# > Ref: https://github.com/ROCm/ROCm/issues/1880 -RUN python${PYTHON_VERSION} -m pip install -U --user pip numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf +# > Ref: https://github.com/ROCm/ROCm/issues/1880#issuecomment-1547838227 +# RUN python${PYTHON_VERSION} -m ensurepip --upgrade +RUN python${PYTHON_VERSION} -m pip install -U --user numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf RUN python${PYTHON_VERSION} -m pip install -U --user keras_preprocessing --no-deps RUN sed -i "s/python3/python${PYTHON_VERSION}/g" build_rocm_python3