Erricia:

[0,1,2,3,4,5])


Oh, renumbered. Isn’t the GPU number in nvidia-smi?
It goes through this way but I get the error(phew :)):
RuntimeError: CUDA out of memory. Tried to allocate 384.00 MiB (GPU 0; 31.75 GiB total capacity; 14.70 GiB already allocated; 378.44 MiB free; 14.76 GiB reserved in total by PyTorch)
It seems the memory error is on GPU 0, which is already has something else running on it. That’s why I didn’t start with 0.
What do you think? Which GPU does GPU 0 refer to here?