Fast CUDA SSIM for PyTorch - Installation Configurator
pip install fussim
| PyTorch | CUDA 11.8 | CUDA 12.1 | CUDA 12.4 | CUDA 12.6 | CUDA 12.8 |
|---|---|---|---|---|---|
| 2.5.1 | ✓ | ✓ | ✓ | - | - |
| 2.6.0 | ✓ | - | ✓ | ✓ | - |
| 2.7.1 | ✓ | - | - | ✓ | ✓ |
| 2.8.0 | - | - | - | ✓ | ✓ |
| 2.9.0 | - | - | - | ✓* | ✓* |
* PyTorch 2.9.0 wheels are Linux-only (Windows has a known PyTorch bug)
Python 3.13 requires PyTorch 2.8.0 or newer