gpus with trVAE model

[ilyakorsunsky]

Hi team,

I am trying to run trVAE with GPUs. I made sure that scArches is able to successfully use GPUs by testing the use_cuda flag in the scVI model. However, I am not certain whether the trVAE model is making use of GPUs or CPUs. Do you have any suggestions for how to make sure that trVAE uses GPUs?

Thanks!

Ilya

[Cottoneyejoe95]

Hi Ilya,

sorry for the late reply. At the moment it is only possible to check that after training by printing the Trainer.device parameter. However if you want to check if the trainer will use cuda you can just run this command upfront:

print(torch.cuda.is_available())

If that returns 'True' the trainer/model will use GPU too.

Best
Marco