We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a8f2c4d commit 0f6b00bCopy full SHA for 0f6b00b
images/runtime/examples/ray-torch-rocm/Dockerfile
@@ -1,9 +1,9 @@
1
-FROM quay.io/modh/ray:2.35.0-py311-rocm61
+FROM quay.io/modh/ray:2.35.0-py311-rocm62
2
3
ENV PYTORCH_ROCM_ARCH=gfx90a;gfx941;gfx942
4
5
# Install PyTorch
6
-RUN python3 -m pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/rocm6.1
+RUN python3 -m pip install torch==2.5.1 --index-url https://download.pytorch.org/whl/rocm6.2
7
8
# Install Flash Attention
9
RUN python3 -m pip install wheel ninja
0 commit comments