fix: build config and depenency

tensorflow.dockerfile

@@ -2,6 +2,7 @@ ARG ROCM_VERSION="5.7.3"
 
 FROM wtongze/rocm-base:${ROCM_VERSION}
 
+ARG ROCM_VERSION="5.7.3"
 ARG TENSORFLOW_BRANCH="r2.12-rocm-enhanced"
 ARG PYTHON_VERSION="3.11"
 ARG NUMPY_VERSION="1.24.3"
@@ -47,11 +48,11 @@ RUN sed -i -e "s/python3/python${PYTHON_VERSION}/g" -e "s/pip3/python${PYTHON_VE
 RUN sudo ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
 
 ENV ROCM_PATH=/opt/rocm-${ROCM_VERSION}
-ENV ROCM_TOOLKIT_PATH=/opt/rocm-${ROCM_VERSION}
-RUN ./build_rocm_python3 /opt/rocm-${ROCM_VERSION}
+ENV ROCM_TOOLKIT_PATH=${ROCM_PATH}
+RUN ./build_rocm_python3 ${ROCM_PATH}
 
 WORKDIR /tmp/tensorflow_pkg
-RUN cp (ls -l tensorflow*.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/tensorflow-rocm.whl
+RUN cp $(ls -l tensorflow*.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/tensorflow-rocm.whl
 
 WORKDIR /home/rocm
 CMD [ "/bin/bash", "-l" ]