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(
18
18
channel : str ,
19
19
) -> List [str ]:
20
20
torch_pkg = "torch"
21
+ download_pytorch_org = "download.pytorch.org"
21
22
if pytorch_version != "" :
22
23
torch_pkg += f"=={ pytorch_version } "
23
24
pip_install = f"pip install { torch_pkg } "
24
25
if channel == "nightly" :
25
26
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 } "
27
30
return [f"export PIP_INSTALL_TORCH='{ pip_install } --index-url { extra_index } '" ]
28
31
29
32
You can’t perform that action at this time.
0 commit comments