File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
tools/pkg-helpers/pytorch_pkg_helpers Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -18,12 +18,15 @@ def get_pytorch_pip_install_command(
1818 channel : str ,
1919) -> List [str ]:
2020 torch_pkg = "torch"
21+ download_pytorch_org = "download.pytorch.org"
2122 if pytorch_version != "" :
2223 torch_pkg += f"=={ pytorch_version } "
2324 pip_install = f"pip install { torch_pkg } "
2425 if channel == "nightly" :
2526 pip_install += " --pre"
26- 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 } "
2730 return [f"export PIP_INSTALL_TORCH='{ pip_install } --index-url { extra_index } '" ]
2831
2932
You can’t perform that action at this time.
0 commit comments