Single GPU Training

[KomputerMaster64]

I wanted to plug and play the model for sampling however I am able to use a single GPU for the task. I wanted to know how to modify the mode for single GPU training.
The following output is shown if I directly run the training file on say Google Colab.

RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method

[VedantDere0104]

use --num_process_per_node 1