Skip to content
Draft
Changes from all commits
Commits
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
326 changes: 324 additions & 2 deletions docs/software/ml/pytorch.md
Original file line number Diff line number Diff line change
Expand Up @@ -255,8 +255,330 @@ The PyTorch uenv is versioned according to the PyTorch version it provides.

| version | node types | system |
|-----------|------------|-------------------------|
| v2.8.0 | gh200 | clariden, daint, santis |
| v2.6.0 | gh200 | clariden, daint |
| v2.9.1 | gh200 | clariden, daint, santis |
| v2.8.0 | gh200 | clariden, daint, santis |
| v2.6.0 | gh200 | clariden, daint |

=== "v2.9.1"

??? info "non-Python packages exposed via the `default` view"

| Package | Version |
|---------------------|------------------|
| `abseil-cpp` | `20250814.1` |
| `autoconf` | `2.72` |
| `automake` | `1.16.5` |
| `aws-ofi-nccl` | `1.17.1` |
| `bc` | `1.07.1` |
| `berkeley-db` | `18.1.40` |
| `binutils` | `2.45` |
| `bison` | `3.8.2` |
| `boost` | `1.89.0` |
| `bzip2` | `1.0.8` |
| `ca-certificates-mozilla` | `2025-08-12` |
| `c-ares` | `1.28.1` |
| `cassini-headers` | `git.59b6de6a91d9637809677c50cc48b607a91a9acb=main` |
| `c-blosc` | `1.21.6` |
| `check` | `0.15.2` |
| `cmake` | `3.31.9` |
| `cpuinfo` | `2025-03-21` |
| `cray-gtl` | `9.0.0` |
| `cray-mpich` | `9.0.0` |
| `cray-pals` | `1.3.2` |
| `cray-pmi` | `6.1.15` |
| `cublasmp` | `0.5.0.898` |
| `cuda` | `12.9.1` |
| `cudnn` | `9.8.0.87-12` |
| `cudss` | `0.7.0` |
| `curl` | `8.15.0` |
| `cusparselt` | `0.8.1-cuda120` |
| `cutensor` | `2.0.1.2` |
| `cutlass` | `4.1.0` |
| `cxi-driver` | `git.08deb056fac4ca8b0d3d39b5f7cc0dad019ee266=main` |
| `diffutils` | `3.12` |
| `dynolog` | `0.5.0` |
| `ed` | `1.4` |
| `eigen` | `5.0.1` |
| `expat` | `2.7.3` |
| `faiss` | `1.8.0` |
| `fftw` | `3.3.10` |
| `findutils` | `4.10.0` |
| `flex` | `2.6.3` |
| `fmt` | `12.1.0` |
| `fp16` | `2020-05-14` |
| `fxdiv` | `2020-04-17` |
| `gcc-runtime` | `14.2.0` |
| `gdbm` | `1.25` |
| `gdrcopy` | `2.5.1` |
| `gettext` | `0.23.1` |
| `git` | `2.48.1` |
| `gmake` | `4.4.1` |
| `gsl` | `2.8` |
| `hdf5` | `1.14.6` |
| `hwloc` | `2.12.2` |
| `hydra` | `4.2.1` |
| `json-c` | `0.18` |
| `krb5` | `1.21.3` |
| `libaec` | `1.1.4` |
| `libaio` | `0.3.113` |
| `libbsd` | `0.12.2` |
| `libconfig` | `1.7.3` |
| `libcxi` | `git.be1f7149482581ad589a124e5f6764b9d20d2d45=main` |
| `libedit` | `3.1-20240808` |
| `libfabric` | `2.3.1` |
| `libffi` | `3.5.2` |
| `libfuse` | `2.9.9` |
| `libgit2` | `1.9.1` |
| `libiconv` | `1.18` |
| `libidn2` | `2.3.7` |
| `libjpeg-turbo` | `3.0.4` |
| `libmd` | `1.1.0` |
| `libnl` | `3.3.0` |
| `libpciaccess` | `0.17` |
| `libpng` | `1.6.47` |
| `libpthread-stubs` | `0.5` |
| `libsigsegv` | `2.14` |
| `libssh2` | `1.11.1` |
| `libtool` | `2.4.7` |
| `libtree` | `3.1.1` |
| `libunistring` | `1.2` |
| `libunwind` | `master` |
| `liburing` | `2.12` |
| `libuv` | `1.48.0` |
| `libxau` | `1.0.12` |
| `libxcb` | `1.17.0` |
| `libxcrypt` | `4.4.38` |
| `libxdmcp` | `1.1.5` |
| `libxml2` | `2.13.5` |
| `libyaml` | `0.2.5` |
| `lm-sensors` | `3-6-0` |
| `lua` | `5.4.6` |
| `lz4` | `1.10.0` |
| `m4` | `1.4.20` |
| `magma` | `2.9.0` |
| `meson` | `1.8.5` |
| `nasm` | `2.16.03` |
| `nccl` | `2.28.9-1` |
| `nccl-tests` | `2.17.6` |
| `ncurses` | `6.5-20250705` |
| `netcdf-c` | `4.9.3` |
| `netcdf-cxx` | `4.2` |
| `netcdf-fortran` | `4.6.2` |
| `nghttp2` | `1.67.1` |
| `ninja` | `1.13.0` |
| `nlohmann-json` | `3.12.0` |
| `numactl` | `2.0.18` |
| `nvidia-mathdx` | `25.06.1-cuda12` |
| `nvshmem` | `3.4.5` |
| `nvtx` | `3.2.1` |
| `openblas` | `0.3.30` |
| `openssh` | `9.9p1` |
| `openssl` | `3.6.0` |
| `osu-micro-benchmarks` | `7.5.1` |
| `papi` | `7.2.0` |
| `patchelf` | `0.17.2` |
| `pcre2` | `10.44` |
| `pcre` | `8.45` |
| `perl` | `5.42.0` |
| `pigz` | `2.8` |
| `pkgconf` | `2.5.1` |
| `prometheus-cpp` | `1.3.0` |
| `protobuf` | `33.0` |
| `psimd` | `2020-05-17` |
| `pthreadpool` | `2023-08-29` |
| `py-absl-py` | `1.4.0` |
| `py-annotated-types` | `0.7.0` |
| `py-build` | `1.2.2` |
| `py-calver` | `2025.4.17` |
| `py-certifi` | `2025.7.14` |
| `py-charset-normalizer` | `3.4.4` |
| `py-coherent-licensed` | `0.5.2` |
| `py-cuda-bindings` | `12.9.0` |
| `py-cuda-core` | `0.4.2` |
| `py-cutlass` | `4.1.0` |
| `py-cython` | `3.1.3` |
| `py-einops` | `0.8.1` |
| `py-filelock` | `3.19.1` |
| `py-flit-core` | `3.12.0` |
| `py-fsspec` | `2025.9.0` |
| `py-grpcio` | `1.75.0` |
| `py-hatch-fancy-pypi-readme` | `25.1.0` |
| `py-hatchling` | `1.27.0` |
| `py-hatch-vcs` | `0.5.0` |
| `py-hf-xet` | `1.2.0` |
| `py-huggingface-hub` | `0.36.0` |
| `py-idna` | `3.10` |
| `py-importlib-metadata` | `8.7.0` |
| `py-iniconfig` | `2.1.0` |
| `py-jinja2` | `3.1.6` |
| `py-lightning-utilities` | `0.11.2` |
| `py-lit` | `18.1.8` |
| `py-markdown` | `3.4.1` |
| `py-markupsafe` | `3.0.2` |
| `py-maturin` | `1.9.6` |
| `py-meson-python` | `0.18.0` |
| `py-ml-dtypes` | `0.5.3` |
| `py-mpi4py` | `4.1.1` |
| `py-mpmath` | `1.3.0` |
| `py-networkx` | `3.5` |
| `py-numpy` | `2.3.5` |
| `py-onnx` | `1.20.0` |
| `py-onnx-ir` | `0.1.12` |
| `py-onnxscript` | `0.5.6` |
| `py-packaging` | `25.0` |
| `py-pathspec` | `0.12.1` |
| `py-pillow` | `12.0.0` |
| `py-pip` | `25.1.1` |
| `py-pluggy` | `1.6.0` |
| `py-protobuf` | `6.33.0` |
| `py-pybind11` | `3.0.1` |
| `py-pyclibrary` | `0.2.2` |
| `py-pydantic` | `2.12.4` |
| `py-pydantic-core` | `2.41.5` |
| `py-pygments` | `2.19.2` |
| `py-pyparsing` | `3.2.5` |
| `py-pyproject-hooks` | `1.2.0` |
| `py-pyproject-metadata` | `0.9.1` |
| `py-pytest` | `9.0.0` |
| `py-pyyaml` | `6.0.3` |
| `py-regex` | `2025.11.3` |
| `py-requests` | `2.32.5` |
| `py-safetensors` | `0.6.2` |
| `py-scikit-build-core` | `0.11.5` |
| `py-semantic-version` | `2.10.0` |
| `py-setuptools` | `79.0.1` |
| `py-setuptools-rust` | `1.12.0` |
| `py-setuptools-scm` | `8.2.1` |
| `py-six` | `1.17.0` |
| `py-sympy` | `1.14.0` |
| `py-tensorboard` | `2.20.0` |
| `py-tensorboard-data-server` | `0.7.0` |
| `python` | `3.12.12` |
| `python-venv` | `1.0` |
| `py-tokenizers` | `0.22.1` |
| `py-torch` | `2.9.1` |
| `py-torchaudio` | `2.9.1` |
| `py-torchmetrics` | `1.8.2` |
| `py-torch-nvidia-apex` | `25.09` |
| `py-torchvision` | `0.24.1` |
| `py-tqdm` | `4.67.1` |
| `py-transformer-engine` | `2.9` |
| `py-transformers` | `4.57.0` |
| `py-triton` | `3.5.1` |
| `py-trove-classifiers` | `2025.9.11.17` |
| `py-typing-extensions` | `4.14.1` |
| `py-typing-inspection` | `0.4.2` |
| `py-urllib3` | `2.5.0` |
| `py-vllm` | `0.12.0` |
| `py-werkzeug` | `3.1.3` |
| `py-wheel` | `0.45.1` |
| `py-zipp` | `3.23.0` |
| `re2` | `2024-07-02` |
| `re2c` | `3.1` |
| `readline` | `8.3` |
| `rust` | `1.91.0` |
| `rust-bootstrap` | `1.91.0` |
| `sed` | `4.9` |
| `sleef` | `3.8` |
| `snappy` | `1.2.1` |
| `sqlite` | `3.50.4` |
| `swig` | `4.1.1` |
| `tar` | `1.35` |
| `texinfo` | `7.2` |
| `ucc` | `1.6.0` |
| `ucx` | `1.19.0` |
| `unzip` | `6.0` |
| `util-linux-uuid` | `2.41` |
| `util-macros` | `1.20.1` |
| `valgrind` | `3.26.0` |
| `which` | `2.21` |
| `xcb-proto` | `1.17.0` |
| `xcb-util` | `0.4.1` |
| `xcb-util-cursor` | `0.1.5` |
| `xcb-util-image` | `0.4.1` |
| `xcb-util-renderutil` | `0.3.10` |
| `xpmem` | `2.9.6` |
| `xproto` | `7.0.31` |
| `xz` | `5.6.3` |
| `zlib` | `1.3.1` |
| `zlib-ng` | `2.2.4` |
| `zstd` | `1.5.7` |

??? info "Python packages exposed via the `default` view"

| Package | Version |
|---------------------|------------------|
| `absl-py` | `1.4.0` |
| `annotated-types` | `0.7.0` |
| `apex` | `0.1` |
| `certifi` | `2025.7.14` |
| `charset-normalizer` | `3.4.4` |
| `cuda-bindings` | `12.9.0` |
| `cuda-core` | `0.4.2` |
| `cutlass_library` | `4.0.0` |
| `Cython` | `3.1.3` |
| `einops` | `0.8.1` |
| `faiss` | `1.8.0` |
| `filelock` | `3.19.1` |
| `fsspec` | `2025.9.0` |
| `grpcio` | `1.75.0` |
| `hf-xet` | `1.2.0` |
| `huggingface_hub` | `0.36.0` |
| `idna` | `3.10` |
| `importlib_metadata` | `8.7.0` |
| `iniconfig` | `2.1.0` |
| `Jinja2` | `3.1.6` |
| `lightning-utilities` | `0.11.2` |
| `Markdown` | `3.4.1` |
| `MarkupSafe` | `3.0.2` |
| `meson` | `1.8.5` |
| `ml_dtypes` | `0.5.3` |
| `mpi4py` | `4.1.1` |
| `mpmath` | `1.3.0` |
| `networkx` | `3.5` |
| `numpy` | `2.3.5` |
| `nvshmem4py-cu12` | `0.1.2` |
| `nvtx` | `0.2.12` |
| `onnx` | `1.20.0` |
| `onnx-ir` | `0.1.12` |
| `onnxscript` | `0.5.6.dev20251209` |
| `packaging` | `25.0` |
| `pillow` | `12.0.0` |
| `pip` | `25.1.1` |
| `pluggy` | `1.6.0` |
| `protobuf` | `6.33.0` |
| `pybind11` | `3.0.1` |
| `pyclibrary` | `0.2.2` |
| `pycute` | `4.0.0` |
| `pydantic` | `2.12.4` |
| `pydantic_core` | `2.41.5` |
| `Pygments` | `2.19.2` |
| `pyparsing` | `3.2.5` |
| `pytest` | `9.0.0` |
| `PyYAML` | `6.0.3` |
| `regex` | `2025.11.3` |
| `requests` | `2.32.5` |
| `safetensors` | `0.6.2` |
| `setuptools` | `79.0.1` |
| `six` | `1.17.0` |
| `sympy` | `1.14.0` |
| `tensorboard` | `2.20.0` |
| `tensorboard_data_server` | `0.7.0` |
| `tokenizers` | `0.22.1` |
| `torch` | `2.9.1` |
| `torchaudio` | `2.9.1+a224ab2` |
| `torchmetrics` | `1.8.2` |
| `torchvision` | `0.24.1` |
| `tqdm` | `4.67.1` |
| `transformer_engine` | `2.9.0+70f5366` |
| `transformers` | `4.57.0` |
| `triton` | `3.5.1` |
| `typing_extensions` | `4.14.1` |
| `typing-inspection` | `0.4.2` |
| `urllib3` | `2.5.0` |
| `vllm` | `0.1.dev1+g4fd9d6a.d19800101` |
| `Werkzeug` | `3.1.3` |
| `zipp` | `3.23.0` |

=== "v2.8.0"

Expand Down