Skip to content

Commit

Permalink
Set PyTorch 2.5.1 as the default version (#2399)
Browse files Browse the repository at this point in the history 
Co-authored-by: Joonas Somero <[email protected]>
  • Loading branch information
@mvsjober @joonas-somero
mvsjober and joonas-somero authored Jan 21, 2025
1 parent ec4f5a3 commit b07e704
Showing 1 changed file with 22 additions and 20 deletions.
42 changes: 22 additions & 20 deletions docs/apps/pytorch.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,8 @@ tags:
Machine learning framework for Python.

!!! info "News"
**20.1.2025** PyTorch 2.5.1 set as the default version on all supercomputers.

**20.12.2024** PyTorch 2.5.1 with ROCm 6.2 added to LUMI. Includes all
the usual packages plus experimental support for FAISS.

Expand Down Expand Up @@ -68,26 +70,26 @@ Machine learning framework for Python.

Currently supported PyTorch versions:

| Version | Module | Puhti | Mahti | LUMI | Notes |
|:--------|----------------|:-----:|:-----:|------|:------------------------------|
| 2.5.1 | `pytorch/2.5` | X | X | X | Tykky-wrappers (Puhti, Mahti) |
| 2.4.1 | `pytorch/2.4` | - | - | X | default version |
| 2.4.0 | `pytorch/2.4` | X | X | - | New tykky-based wrappers |
| 2.3.1 | `pytorch/2.3` | X | X | - | New tykky-based wrappers |
| 2.2.2 | `pytorch/2.2` | - | - | X | default version |
| 2.2.1 | `pytorch/2.2` | X | X | - | |
| 2.1.2 | `pytorch/2.1` | - | - | X | |
| 2.1.0 | `pytorch/2.1` | X | X | - | |
| 2.0.1 | `pytorch/2.0` | - | - | X | |
| 2.0.0 | `pytorch/2.0` | X | X | - | |
| 1.13.1 | `pytorch/1.13` | - | - | X | limited multi-node support |
| 1.13.0 | `pytorch/1.13` | X | X | - | |
| 1.12.0 | `pytorch/1.12` | X | X | - | |
| 1.11.0 | `pytorch/1.11` | X | X | - | |
| 1.10.0 | `pytorch/1.10` | (x) | (x) | - | |
| 1.9.0 | `pytorch/1.9` | (x) | (x) | - | |
| 1.8.1 | `pytorch/1.8` | (x) | (x) | - | |
| 1.7.1 | `pytorch/1.7` | (x) | - | - | |
| Version | Module | Puhti | Mahti | LUMI | Notes |
|:--------|----------------|:-----:|:-----:|------|:---------------------------|
| 2.5.1 | `pytorch/2.5` | X | X | X | Default version |
| 2.4.1 | `pytorch/2.4` | - | - | X | |
| 2.4.0 | `pytorch/2.4` | X | X | - | New tykky-based wrappers |
| 2.3.1 | `pytorch/2.3` | X | X | - | New tykky-based wrappers |
| 2.2.2 | `pytorch/2.2` | - | - | X | |
| 2.2.1 | `pytorch/2.2` | X | X | - | |
| 2.1.2 | `pytorch/2.1` | - | - | X | |
| 2.1.0 | `pytorch/2.1` | X | X | - | |
| 2.0.1 | `pytorch/2.0` | - | - | X | |
| 2.0.0 | `pytorch/2.0` | X | X | - | |
| 1.13.1 | `pytorch/1.13` | - | - | X | limited multi-node support |
| 1.13.0 | `pytorch/1.13` | X | X | - | |
| 1.12.0 | `pytorch/1.12` | X | X | - | |
| 1.11.0 | `pytorch/1.11` | X | X | - | |
| 1.10.0 | `pytorch/1.10` | (x) | (x) | - | |
| 1.9.0 | `pytorch/1.9` | (x) | (x) | - | |
| 1.8.1 | `pytorch/1.8` | (x) | (x) | - | |
| 1.7.1 | `pytorch/1.7` | (x) | - | - | |

Includes [PyTorch](https://pytorch.org/) and related libraries with
GPU support via CUDA/ROCm.
Expand Down

0 comments on commit b07e704

Please sign in to comment.