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.
2 parents e88977e + a7afd24 commit 8fd8a57Copy full SHA for 8fd8a57
.github/workflows/docker/Dockerfile
@@ -7,6 +7,13 @@ ARG PRIMUS_TURBO_COMMIT
7
# Install basic dependencies
8
RUN apt-get update
9
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
+
17
# Clone and install the Primus-Turbo
18
WORKDIR /opt
19
RUN mkdir -p /opt && cd /opt && \
0 commit comments