Fix torch install instructions (#1292)

* Fix torch install instructions * test
pytorch · Jul 23, 2024 · b2a654a · b2a654a
1 parent 775ce42
commit b2a654a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/packaging/pkg_helpers.bash b/packaging/pkg_helpers.bash
@@ -141,8 +141,7 @@ setup_pip_pytorch_version() {
     export PYTORCH_VERSION="$(pip show torch | grep ^Version: | sed 's/Version:  *//' | sed 's/+.\+//')"
   else
     pip_install "torch==$PYTORCH_VERSION$PYTORCH_VERSION_SUFFIX" \
-      -f https://download.pytorch.org/whl/ \
-      -f "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/"
+      --index-url "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${WHEEL_DIR}"
   fi
 }