From 227c721385cbae11f639693b13a2f92dbb90138b Mon Sep 17 00:00:00 2001
From: Tongze Wang <wtongze@gmail.com>
Date: Tue, 10 Sep 2024 14:05:42 -0700
Subject: [PATCH] fix: path

---
 base.dockerfile       | 2 +-
 tensorflow.dockerfile | 2 --
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/base.dockerfile b/base.dockerfile
index c1db6d6..e2c82a6 100644
--- a/base.dockerfile
+++ b/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" ]
diff --git a/tensorflow.dockerfile b/tensorflow.dockerfile
index be61f3e..1db4bbe 100644
--- a/tensorflow.dockerfile
+++ b/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