fix: change arch to gfx803 only

tensorflow.dockerfile

@@ -6,7 +6,8 @@ ARG ROCM_VERSION="5.7.3"
 ARG TENSORFLOW_BRANCH="r2.12-rocm-enhanced"
 ARG PYTHON_VERSION="3.11"
 ARG NUMPY_VERSION="1.24.3"
-ARG HIP_ARCH="gfx803;gfx900"
+# "gfx803" or "gfx803;gfx900"
+ARG HIP_ARCH="gfx803"
 ARG DEBIAN_FRONTEND="noninteractive"
 
 # Add GPU target
@@ -46,7 +47,7 @@ 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/
 
 WORKDIR /home/rocm
 CMD [ "/bin/bash", "-l" ]