Skip to content

Commit

Permalink
Add PyTorch 2.5.0 support
Browse files Browse the repository at this point in the history
  • Loading branch information
MiroPsota committed Oct 28, 2024
1 parent 42e9ec5 commit 6cc4e34
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ pytorch3d==0.7.6+pt2.2.1cu121

## Supported combinations

Tested with PyTorch `1.11.0` - `2.4.1` and their respective compute platforms and supported OSes, with an exception for `cu102` on Windows (no VS 2017 on the GH `windows-2019` runner), and the `rocm` platform.
Tested with PyTorch `1.11.0` - `2.5.0` and their respective compute platforms and supported OSes, with an exception for `cu102` on Windows (no VS 2017 on the GH `windows-2019` runner), and the `rocm` platform.

## Pitfalls

Expand Down
4 changes: 4 additions & 0 deletions get_jobs.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ class TorchRelease:
"2.3.1": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "rocm6.0")),
"2.4.0": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "cu124", "rocm6.1")),
"2.4.1": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "cu124", "rocm6.1")),
"2.5.0": TorchRelease(("3.9", "3.10", "3.11", "3.12", "3.13"), ("cpu", "cu118", "cu121", "cu124", "rocm6.2")),
}


Expand Down Expand Up @@ -76,6 +77,9 @@ def main():
if torch_version in ("1.13.0", "1.13.1") and os_name != LINUX_X64:
continue

if torch_version.startswith("2.5") and python_version == "3.13" and os_name != LINUX_X64:
continue

# Requires VS 2017 not presented in Windows GH runner
if compute_platform == "cu102" and os_name == WINDOWS_X64:
continue
Expand Down

0 comments on commit 6cc4e34

Please sign in to comment.