.cirlceci: Be specific about CU_VERSION for pytorch (#2827)

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 <[email protected]>

Author: seemethere

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
 }