Single GPU

[Hepta-Col]

I can only use a single GPU. Is it possible to modify your code to run on a single GPU? If so, could you tell me where I should modify it? Thank you so much!

[LiJunnan1992]

You should probably be able to run on single GPU using the same command.

[liuhui0401]

I have the same question. If I use a single GPU, the code will warn that 'You have chosen a specific GPU. This will completely disable data parallelism.' And I can't use pdb to debug. I wonder how did you debug? Thanks.