How can I choose my preferred GPU?

[GuGuGuDove]

The trainer will move diffusion (the model) to the accelerator device -- cuda, which is written and fixed in Accelerator class. And accelerator.device is a read-only property that can not be changed.

How can I perform my training on cuda:1 (or other devices)?

Thanks a lot.

[francotheengineer]

run export CUDA_VISIBLE_DEVICES=1 to select gpu 1 in bash or your chosen shell before running your python script. The GPU ID if 0 indexed, so 1 is the 2nd GPU in your machine.`

In your training script set the cuda cuda:0 since on python side it will only see 1 gpu.