叶子🍃
叶子🍃
发布于 2026-06-12 / 7 阅读
0

uv 安装 pytorch 的踩坑经历

环境:

  • NVIDIA Driver: 560.35.03 (最高支持到 CUDA 12.6)

要安装 torch==2.11.0+cu126,走官方指定的方法应该是

uv pip install torch==2.11.0 torchvision torchaudio --index-url https://mirrors.aliyun.com/pytorch-wheels/cu126

但是发现一直找不到package,最终会忽略掉--index-url,走默认源安装了+cu13*的。

原因是uv 会把这个地址当作 PEP 503 simple index 根目录,然后去找类似:

https://mirrors.aliyun.com/pytorch-wheels/cu126/torch/

发现找不到,就fallback到默认源。

正确的方法应该是:

uv pip install `
  "torch==2.11.0+cu126" `
  "torchvision==0.26.0+cu126" `
  "torchaudio==2.11.0+cu126" `
  --find-links https://mirrors.aliyun.com/pytorch-wheels/cu126/ `
  --index-url https://mirrors.aliyun.com/pypi/simple/