Python
/
uv
mirror of https://github.com/astral-sh/uv
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update pytorch.md to include information about supporting CUDA 13.0.x (#16957) 

## Summary

This change updates the guide about integration with pytorch to include
the CUDA 13.0 variant.

## Test Plan

I generated the documentation and verified that it looked correct.
This commit is contained in:
Matthew Woolf 2025-12-03 06:47:57 -08:00 committed by GitHub
parent 1d8252599a
commit 20ab80ad8f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
docs/guides/integration/pytorch.md
View File

@ -103,6 +103,15 @@ In such cases, the first step is to add the relevant PyTorch index to your `pypr
explicit = true explicit = true
``` ```
=== "CUDA 13.0"
```toml
[[tool.uv.index]]
name = "pytorch-cu130"
url = "https://download.pytorch.org/whl/cu130"
explicit = true
```
=== "ROCm6" === "ROCm6"
```toml ```toml
@ -184,6 +193,21 @@ Next, update the `pyproject.toml` to point `torch` and `torchvision` to the desi
] ]
``` ```
=== "CUDA 13.0"
PyTorch doesn't publish CUDA builds for macOS. As such, we gate on `sys_platform` to instruct uv to limit
the PyTorch index to Linux and Windows, falling back to PyPI on macOS:
```toml
[tool.uv.sources]
torch = [
{ index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
torchvision = [
{ index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
```
=== "ROCm6" === "ROCm6"
PyTorch doesn't publish ROCm6 builds for macOS or Windows. As such, we gate on `sys_platform` to instruct uv PyTorch doesn't publish ROCm6 builds for macOS or Windows. As such, we gate on `sys_platform` to instruct uv