'cu124' change to Nightly Torch

YanWenKun · Jul 23, 2024 · fa6514a · fa6514a
1 parent 2f83ac5
commit fa6514a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cu124/Dockerfile b/cu124/Dockerfile
@@ -137,7 +137,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     && pip install \
         --upgrade pip wheel setuptools \
     && pip install \
-        --pre xformers torchvision torchaudio \
+        --pre torch torchvision torchaudio \
         --index-url https://download.pytorch.org/whl/nightly/cu124 \
         --extra-index-url https://pypi.org/simple