...
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://uniupb-paderbornpc2.atlassian.net/wiki/spaces/PC2DOK/pages/129443241902952/Running+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).
...