Skip to content

Update libtorch to 2.9 #1636

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Oct 18, 2025
Merged

Update libtorch to 2.9 #1636

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
03bb18c
Update libtorch to 2.9
stewartj3 Oct 18, 2025
6279bb0
Replace README supported CUDA versions with table
stewartj3 Oct 18, 2025
e4a1bf6
Update parsing for libtorch versions to select correct cuda versions
stewartj3 Oct 18, 2025
192f14a
Add CUDA to the version numbers on the tables
stewartj3 Oct 18, 2025
18fa6de
Apply suggestion from @polvalente
polvalente Oct 18, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions torchx/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,11 +38,11 @@ to `2.4.0` or later.
If you want torch with CUDA support, please use `LIBTORCH_TARGET` to choose
CUDA versions. The current supported targets are:

- `cpu` default CPU only version
- `cu118` CUDA 11.8 and CPU version (no macOS support, libtorch `< 2.8.0` only)
- `cu126` CUDA 12.6 and CPU version (no macOS support)
- `cu128` CUDA 12.8 and CPU version (no macOS support)
- `cu129` CUDA 12.9 and CPU version (no macOS support, libtorch `>= 2.8.0` only)
| Torch Version | Supported Targets |
| --------------- | ---------------------------------------------------- |
| 2.7.0 | cpu, cu118 (CUDA 11.8), cu126 (12.6), cu128 (12.8) |
Copy link
Contributor

@polvalente polvalente Oct 18, 2025 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
| 2.7.0 | cpu, cu118 (CUDA 11.8), cu126 (12.6), cu128 (12.8) |
| < 2.8.0 | cpu, cu118 (CUDA 11.8), cu126 (12.6), cu128 (12.8) |

Copy link
Contributor

@polvalente polvalente Oct 18, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for the 3 lines all parenthesis should say CUDA x instead of the first one. I don't seem to be able to apply suggestions so I can edit this by myself

Copy link
Contributor Author

@stewartj3 stewartj3 Oct 18, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added CUDA to the other versions in the table.

| 2.8.0 | cpu, cu126 (CUDA 12.6), cu128 (12.8), cu129 (12.9) |
| 2.9.0 | cpu, cu126 (CUDA 12.6), cu128 (12.8), cu130 (13.0) |

Once downloaded, we will compile `Torchx` bindings. You will need `make`/`nmake`,
`cmake` (3.12+) and a `C++` compiler. If building on Windows, you will need:
Expand Down
22 changes: 14 additions & 8 deletions torchx/mix.exs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -75,21 +75,27 @@ defmodule Torchx.MixProject do

defp libtorch_config() do
target = System.get_env("LIBTORCH_TARGET", "cpu")
version = System.get_env("LIBTORCH_VERSION", "2.8.0")
version = System.get_env("LIBTORCH_VERSION", "2.9.0")
env_dir = System.get_env("LIBTORCH_DIR")

# 2.8.0 is the first version that supports cu129 and drops cu118
# cu118 might still be needed for older hardware, so we're keeping it
# for now.
# Supported targets for each LibTorch version:
# 2.7.0: cpu, cu118 (CUDA 11.8), cu126 (12.6), cu128 (12.8)
# 2.8.0: cpu, cu126 (CUDA 12.6), cu128 (12.8), cu129 (12.9)
# 2.9.0: cpu, cu126 (CUDA 12.6), cu128 (12.8), cu130 (13.0)
valid_targets = ["cpu", "cu118", "cu126", "cu128"]

valid_targets =
case Version.parse(version) do
{:ok, parsed} ->
if Version.match?(parsed, "<= 2.7.0") do
valid_targets
else
(valid_targets -- ["cu118"]) ++ ["cu129"]
cond do
Version.match?(parsed, "< 2.8.0") ->
valid_targets

Version.match?(parsed, "< 2.9.0") ->
(valid_targets -- ["cu118"]) ++ ["cu129"]

true ->
(valid_targets -- ["cu118"]) ++ ["cu130"]
end

_ ->
Expand Down