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