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正常
- 执行
uv sync,确认无报错 - 激活虚拟环境:
.\.venv\Scripts\activate - 验证PyTorch CUDA可用性:
import torch print(torch.cuda.is_available()) # 输出True表示CUDA正常工作
内容的提问来源于stack exchange,提问作者Sebastian H
相关产品推荐
相关产品推荐

