[ci]: Update ubuntu and pytorch

[reneleonhardt]

ubuntu-20.04 is not supported anymore: https://github.com/actions/runner-images/pull/11641 https://github.com/state-spaces/mamba/blob/a6a1dae6efbf804c9944a0c2282b437deb4886d8/.github/workflows/publish.yaml#L43-L45

By the way, pyproject.toml allows any version, but CI isn't compatible with torch 2.6 and 2.7? https://github.com/state-spaces/mamba/blob/a6a1dae6efbf804c9944a0c2282b437deb4886d8/.github/workflows/publish.yaml#L47

[tridao]

We've just updated them

[JayTaoHong]

We've just updated them

May I ask what the versions of Ubuntu, python, torch, cuda, causal-conv1d and mamba-ssm are respectively? My current environment is as follows: (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# pip list | grep mamba-ssm mamba-ssm 2.2.5 (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# pip list | grep causal causal-conv1d 1.5.2 (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# pip list | grep torch torch 2.7.1+cu128 torchaudio 2.7.1+cu128 torchvision 0.22.1+cu128 (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# pip list | grep cuda nvidia-cuda-cupti-cu12 12.8.57 nvidia-cuda-nvrtc-cu12 12.8.61 nvidia-cuda-runtime-cu12 12.8.57 (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# pip list | grep triton triton 3.3.1 (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# pip list | grep transformers transformers 4.54.1 (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# lsb_release -a No LSB modules are available. Distributor ID: Ubuntu Description: Ubuntu 20.04.2 LTS Release: 20.04 Codename: focal The error report is as follows: (mamba_cuda12_8) root@jay1:/opt/data/private/codes/sat2rad_ht/src/mamba# python Python 3.10.18 (main, Jun 5 2025, 13:14:17) [GCC 11.2.0] on linux Type "help", "copyright", "credits" or "license" for more information.

import torch from mamba_ssm import Mamba2 Traceback (most recent call last): File "", line 1, in File "/opt/data/private/codes/sat2rad_ht/src/mamba/mamba_ssm/init.py", line 3, in from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn File "/opt/data/private/codes/sat2rad_ht/src/mamba/mamba_ssm/ops/selective_scan_interface.py", line 20, in import selective_scan_cuda ImportError: /usr/lib/x86_64-linux-gnu/libc.so.6: version `GLIBC_2.32' not found (required by /opt/conda/envs/mamba_cuda12_8/lib/python3.10/site-packages/selective_scan_cuda.cpython-310-x86_64-linux-gnu.so) Could you please provide you with the complete download process of mamba2? Thank you very much

[reneleonhardt]

Ubuntu 20.04 support ended in May, it used glibc 2.31. https://launchpad.net/ubuntu/focal/+source/glibc