From d7d26cbba5e88702c61eec5544c5f880c7fc4a72 Mon Sep 17 00:00:00 2001 From: Eli Uriegas Date: Fri, 16 Oct 2020 09:53:24 -0700 Subject: [PATCH] .cirlceci: Be specific about CU_VERSION for pytorch ROCM builds were being picked up for CUDA 10.2 builds since they do not specify a version suffix. We can utilize the CU_VERSION to be specific about which index to actually pull from. Signed-off-by: Eli Uriegas --- packaging/pkg_helpers.bash | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/packaging/pkg_helpers.bash b/packaging/pkg_helpers.bash index 1e9dd90a44d..09d3ddbbecc 100644 --- a/packaging/pkg_helpers.bash +++ b/packaging/pkg_helpers.bash @@ -234,9 +234,9 @@ setup_pip_pytorch_version() { fi else pip_install "torch==$PYTORCH_VERSION$PYTORCH_VERSION_SUFFIX" \ - -f https://download.pytorch.org/whl/torch_stable.html \ - -f https://download.pytorch.org/whl/test/torch_test.html \ - -f https://download.pytorch.org/whl/nightly/torch_nightly.html + -f "https://download.pytorch.org/whl/${CU_VERSION}/torch_stable.html" \ + -f "https://download.pytorch.org/whl/test/${CU_VERSION}/torch_test.html" \ + -f "https://download.pytorch.org/whl/nightly/${CU_VERSION}/torch_nightly.html" fi }