Update Dockerfile CUDA (huggingface#2682)

* Update Dockerfile CUDA * Apply suggestions from code review
referri · Mar 15, 2023 · 9d1341d · 9d1341d
1 parent 4553c29
commit 9d1341d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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 \
@@ -40,4 +39,4 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \
         tensorboard \
         transformers
 
-CMD ["/bin/bash"]
+CMD ["/bin/bash"]