fix: change arch to gfx803 only

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

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"
 

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

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" ]