Skip to content

Commit

Permalink
Dockerfile.rocm.ubi: rollback to UBI 9.4
Browse files Browse the repository at this point in the history
amdgpu builds for 9.5 are not published yet for ROCm 6.2.3, we can't
bump ROCm to 6.3 either because torch nightlies for 6.3 aren't out yet.
  • Loading branch information
dtrifiro committed Dec 19, 2024
1 parent 30d47f8 commit 31fcf13
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile.rocm.ubi
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
## Global Args ##################################################################
ARG BASE_UBI_IMAGE_TAG=9.5-1733767867
ARG BASE_UBI_IMAGE_TAG=9.4
ARG PYTHON_VERSION=3.12
# Default ROCm ARCHes to build vLLM for.
ARG PYTORCH_ROCM_ARCH="gfx908;gfx90a;gfx942;gfx1100"
Expand Down

0 comments on commit 31fcf13

Please sign in to comment.