Cherry pick #1292 (#1293)

atalman · web-flow · commit cce4a8cf4077 · 2024-07-23T17:33:05.000-04:00
* Fix torch install instructions

* test
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
 }
 

Original file line number	Diff line number	Diff line change
`@@ -141,8 +141,7 @@ setup_pip_pytorch_version() {`
`141`	`141`	`export PYTORCH_VERSION="$(pip show torch \| grep ^Version: \| sed 's/Version: *//' \| sed 's/+.\+//')"`
`142`	`142`	`else`
`143`	`143`	`pip_install "torch==$PYTORCH_VERSION$PYTORCH_VERSION_SUFFIX" \`
`144`		`- -f https://download.pytorch.org/whl/ \`
`145`		`- -f "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/"`
	`144`	`+ --index-url "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${WHEEL_DIR}"`
`146`	`145`	`fi`
`147`	`146`	`}`
`148`	`147`