(base) user_name@lipn-tal-labex:~$ srun python3 gpu_test‧py
<torch‧cuda‧device object at 0x7f29f0602d10>
<torch‧cuda‧device object at 0x7f29f0602d90>
<torch‧cuda‧device object at 0x7f29f0602e90>
<torch‧cuda‧device object at 0x7f29f0618cd0>
<torch‧cuda‧device object at 0x7f29f0618d10>
<torch‧cuda‧device object at 0x7f29f0618d90>
<torch‧cuda‧device object at 0x7f29f0618dd0>
<torch‧cuda‧device object at 0x7f29f0618e10>