--extra-index-url

Lightning-AI · Jan 8, 2025 · 7c80d11 · 7c80d11
1 parent 1617b82
commit 7c80d11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dockers/ubuntu-cuda/Dockerfile b/dockers/ubuntu-cuda/Dockerfile
@@ -92,7 +92,7 @@ RUN \
     python assistant.py replace_str_requirements "onnxruntime" "onnxruntime_gpu" --req_files requirements/audio.txt && \
     # install develop environment
     pip install --no-cache-dir -r requirements/_devel.txt \
-      --find-links="https://download.pytorch.org/whl/cu${CU_VERSION_MM}/torch_stable.html" \
+      --extra-index-url="https://download.pytorch.org/whl/cu${CU_VERSION_MM}/" \
       --extra-index-url="https://download.pytorch.org/whl/test/cu${CU_VERSION_MM}" && \
     rm -rf requirements/