fix: change arch to gfx803 only

2024-09-13 09:32:19 -07:00
parent af09aa8e34
commit 0abbf5d2e5
4 changed files with 9 additions and 6 deletions
@@ -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