...
If you would like to use GPUs for development/testing, it is recommended to do that within an interactive Slurm session on the DGX. More information on that can be found here https://upb-pc2.atlassian.net/wiki/spaces/PC2DOK/pages/1902952/Running Compute Jobs+Compute+Jobs#Using-GPUs-for-Development-and-Testing-Purposes
Please keep in mind that you also need to setup the Python environment in an interactive session as written in step 1 (see above).
...