Actually use the GPU when available

Demo process: https://arkindex.teklia.com/process/b162a959-01f8-4f19-aa68-0ff88296f314/0

Warning is

/usr/local/lib/python3.10/site-packages/pytorch_lightning/utilities/distributed.py:50: UserWarning: GPU available but not used. Set the --gpus flag when calling the script.

Update TrainerArgs to set gpus=1 when torch.cuda.is_available() is True.

We will most surely need to update the docker image used. Try nvidia/cuda:12.2.0-base-ubuntu22.04

Edited Apr 18, 2024 by Yoann Schneider

To upload designs, you'll need to enable LFS and have an admin enable hashed storage. More information