Thanks for the great framework you implemented. Here's a small issue. I want to run the program on one GPU while the program calls all GPUs. I tried to set

accelerator="cuda:0"

but errors were raised. Any solutions except setting "CUDA_VISIBLE_DIVICE" in the command line?

Hi! We also use the CUDA_VISIBLE_DIVICE flag before the python script, is the easiest option