I am trying to use multiple GPUs for a code that is already running correctly for one GPU. I did those modifications to make it able to di multiple GPUs:
clas
