使用阿里云镜像下载Torch的uv工具

背景

在深度学习和人工智能领域，PyTorch是一个广泛使用的库。然而，从pytorch.org下载Torch有时可能会遇到速度慢的问题，尤其是在网络环境不佳的地区。为了解决这个问题，uv提供了一个功能，可以自动选择GPU后端进行下载。但是，默认情况下，它使用的是pytorch.org的镜像，且无法更改base。为了提高下载速度，我们可以修改uv的代码，使其使用阿里云镜像来下载Torch。

功能和用法

这个修改后的工具名为 uv-torch-mirror.exe，其使用方法如下：

Usage: uv-torch-mirror.exe <BACKEND> <PACKAGES_AND_ARGS>...

Arguments:
  <BACKEND>               Backend to use, for example: auto, cpu, cu130, xpu, rocm7.2
  <PACKAGES_AND_ARGS>...  Arguments forwarded verbatim to `uv pip install`

Options:
  -h, --help  Print help

Examples:
uv-torch-mirror.exe torch==2.10 torchvision

其中 <BACKEND> 参数用于指定后端，例如 auto、cpu、cu130、xpu、rocm7.2 等，而 <PACKAGES_AND_ARGS> 参数则将直接传递给 uv pip install 命令。

链接

您可以通过以下链接下载该工具：

uv-torch-mirror.exe下载链接

注意

由于阿里云镜像没有提供符合PEP 714/PEP 658的meta文件，同时range读取文件很慢，运行时在 torch==2.9.0+cu129 阶段可能会卡顿1~2分钟。如果您遇到这种情况，请耐心等待。

完整话题阅读

如果您想了解更多信息，可以通过以下链接：

(https:///t/topic/2368311)

来源

本信息来源于 - ，作者是 flt。

使用阿里云镜像下载Torch的uv工具