We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eeced40 commit 333d432Copy full SHA for 333d432
tools/pkg-helpers/pytorch_pkg_helpers/wheel.py
@@ -18,12 +18,15 @@ def get_pytorch_pip_install_command(
18
channel: str,
19
) -> List[str]:
20
torch_pkg = "torch"
21
+ download_pytorch_org = "download.pytorch.org"
22
if pytorch_version != "":
23
torch_pkg += f"=={pytorch_version}"
24
pip_install = f"pip install {torch_pkg}"
25
if channel == "nightly":
26
pip_install += " --pre"
- extra_index = f"https://download.pytorch.org/whl/{channel}/{gpu_arch_version}"
27
+ # Use Meta CDN for nightly builds
28
+ download_pytorch_org = "d3kup0pazkvub8.cloudfront.net"
29
+ extra_index = f"https://{download_pytorch_org}/whl/{channel}/{gpu_arch_version}"
30
return [f"export PIP_INSTALL_TORCH='{pip_install} --index-url {extra_index}'"]
31
32
0 commit comments