search

Alexanders101 / SPANet

Symmetry Preserving Attention Networks for Event Reconstruction
BSD 3-Clause "New" or "Revised" License
36 stars 33 forks source link

SPANet v2, pytorch-lightning 1.9 GPU detection error #6

Open amunozna opened 1 year ago

amunozna commented 1 year ago

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

Alexanders101 commented 1 year ago

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?