Skip to content

Commit 8fd8a57

Browse files
authored
Merge branch 'main' into dev/xiaobo
2 parents e88977e + a7afd24 commit 8fd8a57

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

.github/workflows/docker/Dockerfile

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,13 @@ ARG PRIMUS_TURBO_COMMIT
77
# Install basic dependencies
88
RUN apt-get update
99

10+
RUN pip3 install fbgemm-gpu && \
11+
pip3 install torch==2.10.0.dev20251019+rocm7.0 \
12+
torchao==0.15.0.dev20251015+rocm7.0 \
13+
torchvision==0.25.0.dev20251020+rocm7.0 \
14+
--index-url https://download.pytorch.org/whl/nightly/rocm7.0 \
15+
--force-reinstall
16+
1017
# Clone and install the Primus-Turbo
1118
WORKDIR /opt
1219
RUN mkdir -p /opt && cd /opt && \

0 commit comments

Comments
 (0)