diff --git a/dockers/ubuntu-cuda/Dockerfile b/dockers/ubuntu-cuda/Dockerfile index a93277a2926..39c9953c72e 100644 --- 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/