To add my docker image use as singularity in lxplus
Setting with cuda 12.4, torch 2.4.0, torch-geometry