Using a ROCm6.2.2 base image with torch version that doesn't support …

…aotriton so that the base image won't have it and we can rebuild torch
ROCm · Nov 2, 2024 · 261f8ba · 261f8ba
1 parent 733f79a
commit 261f8ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -1,11 +1,11 @@
 # default base image
-ARG BASE_IMAGE="rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.3.0"
+ARG BASE_IMAGE="rocm/pytorch:rocm6.2.2_ubuntu20.04_py3.9_pytorch_release_2.1.2"
 
 ARG COMMON_WORKDIR=/app
 
 # The following ARGs should be "0" or "1". If "1", the respective component will be built and installed on top of the base image
 ARG BUILD_HIPBLASLT="0"
-ARG BUILD_RCCL="1"
+ARG BUILD_RCCL="0"
 ARG BUILD_FA="1"
 ARG BUILD_TRITON="1"
 ARG BUILD_PYTORCH="1"