Skip to content

Commit a76bb14

Browse files
committed
add extra indexes for the pytorch cpu-only wheel downloads
1 parent 4218c59 commit a76bb14

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

examples/platform_specific_deps/MODULE.bazel

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,10 @@ python.toolchain(
2323
pip = use_extension("@rules_python//python/extensions:pip.bzl", "pip")
2424
pip.parse(
2525
download_only = True,
26+
experimental_extra_index_urls = [
27+
# Add an extra index to query packages
28+
"https://download.pytorch.org/whl/cpu",
29+
],
2630
experimental_index_url = "https://pypi.org/simple", # use Bazel downloader
2731
hub_name = "pip",
2832
python_version = "3.11",

0 commit comments

Comments
 (0)