SPANet v2, pytorch-lightning 1.9 GPU detection error

[amunozna]

Description

Having followed all installation instructions, using environment.yaml file, and after starting training of the ttbar example:
python -m spanet.train -of options_files/full_hadronic_ttbar/example.json --gpus 1
with 1 GPU, an error appeared on the console:
File ".../environment/lib/python3.10/site-packages/pytorch_lightning/trainer/connectors/accelerator_connector.py", line 538, in _choose_gpu_accelerator_backend raise MisconfigurationException("No supported gpu backend found!") lightning_fabric.utilities.exceptions.MisconfigurationException: No supported gpu backend found!

Hardware used

• 1 CPU
• 1 GPU (Cuda version: 12.0)
• 32 GB RAM

[Alexanders101]

It appears that torch still doesn't officially support CUDA 12. The latest version updates torch to 2.0.1 which supports up to 11.8 officially and possibly 12. Could you verify that the newer version works on your GPU?