Update Dockerfile CUDA (#2682)

* Update Dockerfile CUDA * Apply suggestions from code review

Update Dockerfile CUDA (#2682)
* Update Dockerfile CUDA * Apply suggestions from code review
9d1341d6 · Patrick von Platen · GitHub · 4553c29d · 9d1341d6
Unverified Commit 9d1341d6 authored Mar 15, 2023 by Patrick von Platen Committed by GitHub Mar 15, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

docker/diffusers-pytorch-cuda/Dockerfile docker/diffusers-pytorch-cuda/Dockerfile +1 -2

No files found.
--- a/docker/diffusers-pytorch-cuda/Dockerfile
+++ b/docker/diffusers-pytorch-cuda/Dockerfile
@@ -27,7 +27,6 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \
        torch \
        torchvision \
        torchaudio \
-        --extra-index-url https://download.pytorch.org/whl/cu117 && \
    python3 -m pip install --no-cache-dir \
        accelerate \
        datasets \