fix: path

base.dockerfile

@@ -86,7 +86,7 @@ EOF
 # Change to non-root user
 USER rocm
 WORKDIR /home/rocm
-ENV PATH="${PATH}:/opt/rocm/bin"
+ENV PATH="${PATH}:/opt/rocm/bin:/home/rocm/.local/bin"
 
 # Default to login shell
 CMD [ "bash", "-l" ]

tensorflow.dockerfile

@@ -32,8 +32,6 @@ sudo apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev
 sudo rm -rf /var/lib/apt/lists/*
 EOF
 
-ENV PATH=$PATH:/home/rocm/.local/bin
-
 # 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