Skip to content

Commit 0f6b00b

Browse files
RHOAIENG-22169 - Update ROCM example image
1 parent a8f2c4d commit 0f6b00b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

images/runtime/examples/ray-torch-rocm/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
FROM quay.io/modh/ray:2.35.0-py311-rocm61
1+
FROM quay.io/modh/ray:2.35.0-py311-rocm62
22

33
ENV PYTORCH_ROCM_ARCH=gfx90a;gfx941;gfx942
44

55
# Install PyTorch
6-
RUN python3 -m pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/rocm6.1
6+
RUN python3 -m pip install torch==2.5.1 --index-url https://download.pytorch.org/whl/rocm6.2
77

88
# Install Flash Attention
99
RUN python3 -m pip install wheel ninja

0 commit comments

Comments
 (0)