diff --git a/base.dockerfile b/base.dockerfile
index 26dab5c..419e9b7 100644
--- a/base.dockerfile
+++ b/base.dockerfile
@@ -5,8 +5,8 @@ FROM ubuntu:22.04 AS base
 ARG ROCM_VERSION="5.7.3"
 ARG ROCM_APT_CODE_NAME="jammy"
 ARG ROCBLAS_BRANCH="release/rocm-rel-5.7"
-# gfx803 as single target may cause error
-ARG ROCBLAS_ARCH="gfx803;gfx900"
+# "gfx803" or "gfx803;gfx900"
+ARG ROCBLAS_ARCH="gfx803"
 ARG DEBIAN_FRONTEND="noninteractive"
 
 # Install essentials
diff --git a/blender.dockerfile b/blender.dockerfile
index 3431141..14a6245 100644
--- a/blender.dockerfile
+++ b/blender.dockerfile
@@ -2,7 +2,8 @@ ARG ROCM_VERSION="5.7.3"
 
 FROM wtongze/rocm-base:${ROCM_VERSION}
 
-ARG HIP_ARCH="gfx803;gfx900"
+# "gfx803" or "gfx803;gfx900"
+ARG HIP_ARCH="gfx803"
 ARG BLENDER_BRANCH="v3.6.15"
 ARG DEBIAN_FRONTEND="noninteractive"
 
diff --git a/pytorch.dockerfile b/pytorch.dockerfile
index 7e38458..10362d5 100644
--- a/pytorch.dockerfile
+++ b/pytorch.dockerfile
@@ -7,7 +7,8 @@ ARG PYTORCH_BRANCH="v2.2.2"
 ARG TORCHVISION_BRANCH="v0.17.2"
 ARG PYTHON_VERSION="3.11"
 ARG NUMPY_VERSION="1.26.4"
-ARG HIP_ARCH="gfx803;gfx900"
+# "gfx803" or "gfx803;gfx900"
+ARG HIP_ARCH="gfx803"
 ARG DEBIAN_FRONTEND="noninteractive"
 
 # Add GPU target
diff --git a/tensorflow.dockerfile b/tensorflow.dockerfile
index 5af6f84..f548ce7 100644
--- a/tensorflow.dockerfile
+++ b/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" ]