diff --git a/tensorflow.dockerfile b/tensorflow.dockerfile
index d167453..acca18f 100644
--- a/tensorflow.dockerfile
+++ b/tensorflow.dockerfile
@@ -2,9 +2,9 @@ ARG ROCM_VERSION="5.7.3"
 
 FROM wtongze/rocm-base:${ROCM_VERSION}
 
-ARG TENSORFLOW_BRANCH="r2.11-rocm-enhanced"
+ARG TENSORFLOW_BRANCH="r2.12-rocm-enhanced"
 ARG PYTHON_VERSION="3.11"
-ARG NUMPY_VERSION="1.22.4"
+ARG NUMPY_VERSION="1.24.3"
 ARG HIP_ARCH="gfx803;gfx900"
 ARG DEBIAN_FRONTEND="noninteractive"
 
@@ -17,6 +17,14 @@ RUN sudo wget -O /usr/local/bin/bazel https://github.com/bazelbuild/bazelisk/rel
 WORKDIR /home/rocm
 RUN git clone --depth 1 https://github.com/ROCm/tensorflow-upstream.git -b ${TENSORFLOW_BRANCH} tensorflow
 
+# For ROCm 4.5+
+RUN <<EOF
+sudo apt-get update
+sudo apt-get install -y rocm-ml-sdk rocm-hip-sdk
+sudo apt-get clean
+sudo rm -rf /var/lib/apt/lists/*
+EOF
+
 RUN <<EOF
 sudo apt-get update
 sudo apt install -y software-properties-common
@@ -31,14 +39,16 @@ WORKDIR /home/rocm/tensorflow
 
 # TODO: simplify dependencies
 # > Ref: https://github.com/ROCm/ROCm/issues/1880#issuecomment-1547838227
-# RUN python${PYTHON_VERSION} -m ensurepip --upgrade
+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
+RUN sed -i -e "s/python3/python${PYTHON_VERSION}/g" -e "s/pip3/python${PYTHON_VERSION} -m pip/g" build_rocm_python3
 RUN sudo ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
 
-RUN ./build_rocm_python3 /opt/rocm
+ENV ROCM_PATH=/opt/rocm-${ROCM_VERSION}
+ENV ROCM_TOOLKIT_PATH=/opt/rocm-${ROCM_VERSION}
+RUN ./build_rocm_python3 /opt/rocm-${ROCM_VERSION}
 
 WORKDIR /tmp/tensorflow_pkg
 RUN cp (ls -l tensorflow*.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/tensorflow-rocm.whl