Skip to content

Commit a42b6ff

Browse files
committed
[ci][utils.sh] Look on pytorch servers when installing pip deps
When installing the executorch pip package for CI jobs, look on the pytorch servers when resolving dependencies. This lets the executorch package depend on pytorch pre-release and nightly versions.
1 parent 2a89416 commit a42b6ff

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

.ci/scripts/utils.sh

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,11 @@ retry () {
1818
install_executorch() {
1919
which pip
2020
# Install executorch, this assumes that Executorch is checked out in the
21-
# current directory
22-
pip install . --no-build-isolation -v
21+
# current directory. The --extra-index-url options tell pip to look on the
22+
# pytorch servers for nightly and pre-release versions of torch packages.
23+
pip install . --no-build-isolation -v \
24+
--extra-index-url https://download.pytorch.org/whl/test/cpu \
25+
--extra-index-url https://download.pytorch.org/whl/nightly/cpu
2326
# Just print out the list of packages for debugging
2427
pip list
2528
}

0 commit comments

Comments
 (0)