Skip to content

Commit 8270b21

Browse files
committed
pin torch version
1 parent b205fa1 commit 8270b21

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

docker/amd-docker.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ RUN sudo apt update -y \
4242

4343
RUN sudo pip install --upgrade pip
4444

45-
RUN sudo pip install --no-cache-dir pytorch-triton-rocm torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
45+
RUN sudo pip install --no-cache-dir torch==2.10.0.dev20250916+rocm6.3 pytorch-triton-rocm --index-url https://download.pytorch.org/whl/nightly/rocm6.3
4646

4747
RUN git clone --recursive https://github.com/ROCm/aiter.git \
4848
&& cd aiter \

0 commit comments

Comments
 (0)