x

Lightning-AI · Jun 28, 2024 · e12ea4c · e12ea4c
1 parent 5f27fa6
commit e12ea4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile
@@ -92,7 +92,7 @@ RUN \
       -r requirements/pytorch/test.txt \
       -r requirements/pytorch/strategies.txt \
       --find-links="https://download.pytorch.org/whl/cu${CUDA_VERSION_MM//'.'/''}/torch_stable.html" \
-      --find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/torch_test.html"
+      --find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/torch"
 
 RUN \
     # Show what we have