whisperx

[laraPPr]

• link to support ticket: #2025102760001397
• website: https://github.com/m-bain/whisperX
• installation docs: https://github.com/easybuilders/easybuild-easyconfigs/blob/develop/easybuild/easyconfigs/w/WhisperX/WhisperX-3.3.1-foss-2023a-CUDA-12.1.1.eb
• toolchain: if you get it working with foss/2023a fine otherwise we'll have to wait for:
  {ai}[foss/2023b] PyTorch v2.3.0 w/ CUDA 12.4.0 easybuilders/easybuild-easyconfigs#23553 or
  {ai}[foss/2024a] PyTorch v2.9.1 w/ CUDA 12.6.0 easybuilders/easybuild-easyconfigs#24365 or
  {ai}[foss/2024a] PyTorch v2.7.1 w/ CUDA 12.6.0 easybuilders/easybuild-easyconfigs#23923
• easyblock to use: PythonBundle
• required dependencies:
  • ...
• optional dependencies:
  • ...
• notes:
  • Use same toolchain as for pyannote.audio4
• effort: (TBD)
• other install methods
  • conda: yes (link?) / no
  • container image: yes (link?) / no
  • pre-built binaries (RHEL8 Linux x86_64): yes (link?) / no
  • easyconfig outside EasyBuild: yes / no

[pavelToman]

There is PyTorch-2.6.0 for 13.3.0 based toolchain, but no CUDA version - no GPU version of WhisperX. The compatible version of WhisperX is probably v3.4.3 where a dep "torch>=2.5.1" is. Since v3.5.0 there is "torch>=2.7.1" in deps.

Another way is to prepare all the deps with PyTorch-2.7.1-CUDA-12.6.0 - the PR seems good: easybuilders/easybuild-easyconfigs#23923
With this torch I can go for v3.7.3 of WhisperX.

Latest version of WhisperX-3.7.4 requires PyTorch >= 2.8.0.

[pavelToman]

Also I can not go for WhisperX-3.7.3 - it was yanked from pypi.
So I will go for the version 3.7.2.
First I will try v3.7.4 with unpin deps