You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Win10环境下使用uv安装带CuDNN支持的PyTorch遇故障求助

在Win10上用uv正确安装PyTorch(CUDA 12.4)

先解决uv版本不一致问题

版本不匹配是多数命令/配置失效的根源,先统一uv版本:

  • 卸载当前旧版本:uv self uninstall
  • 重新安装最新稳定版:powershell -c "irm https://astral.sh/uv/install.ps1 | iex"
  • 验证版本:uv version,确保和安装提示的版本一致

正确添加PyTorch包(支持uv sync)

不要用uv pip install(临时安装,不会写入pyproject.toml,sync时会被清理),也不要错误配置sources,正确步骤:

方法1:用uv add指定索引(推荐)

uv 0.5+支持给包指定索引,命令如下:

uv add torch torchvision torchaudio --index https://download.pytorch.org/whl/cu124

注意:uv的参数是--index,而非pip的--index-url,0.5+版本已统一该参数命名

执行后会自动写入pyproject.toml,生成的配置格式如下:

[project]
dependencies = [
    "torch",
    "torchvision",
    "torchaudio",
]

[[tool.uv.dependency-sources]]
dependencies = ["torch", "torchvision", "torchaudio"]
index = "https://download.pytorch.org/whl/cu124"

方法2:手动编辑pyproject.toml(适配0.4.x旧版本)

如果暂时无法升级uv到0.5+,用[[tool.uv.index]]配置索引(注意是数组格式):

[project]
dependencies = [
    "torch",
    "torchvision",
    "torchaudio",
]

[[tool.uv.index]]
url = "https://download.pytorch.org/whl/cu124"
priority = "high"

编辑完成后执行uv sync,uv会优先从PyTorch的CUDA索引拉取包。

之前的[tool.uv.sources]配置报错的原因:该字段用于指定单个包的源码URL(如Git仓库或单个whl文件),而非索引地址,因此指向PyTorch索引URL会提示“不是whl文件”,属于用法错误。

验证安装和sync正常

  1. 执行uv sync,确认无报错
  2. 激活虚拟环境:.\.venv\Scripts\activate
  3. 验证PyTorch CUDA可用性:
import torch
print(torch.cuda.is_available())  # 输出True表示CUDA正常工作

内容的提问来源于stack exchange,提问作者Sebastian H

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.16 03:22:42