different way to install PyTorch & xFormers

YanWenKun · Oct 12, 2023 · 6c77abc · 6c77abc
1 parent 2b258c9
commit 6c77abc
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -15,15 +15,10 @@ RUN --mount=type=cache,target=/var/cache/zypp \
         shadow git aria2 \
         Mesa-libGL1
 
-# Install PyTorch (stable version)
+# Install xFormers (dev version, will install PyTorch as well)
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install --break-system-packages \
-        torch torchvision --index-url https://download.pytorch.org/whl/cu118
-
-# Install xFormers (stable version)
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install --break-system-packages \
-        xformers
+        --pre -U xformers
 
 # Deps for main app
 RUN --mount=type=cache,target=/root/.cache/pip \