Allow specifying device to use when training

Closes #139 (closed)

added P2 Quick Win labels

assigned to @mblanco

Tested with this config on straton:

use_ddp	force	use
false	null	GPU: 0
	"cpu"	CPU
	"cuda"	GPU: 0
	"cuda:1"	GPU: 1
true	null	GPU: 0 & 1
	"cpu"	CPU
	"cuda"	GPU: 0 & 1
	"cuda:1"	GPU: 0 & 1

I'm not sure about the behaviour of the last line use_ddp=true and force="cuda:1". In this case, should we just use GPU:1?

requested review from @yschneider

removed review request for @yschneider

added 1 commit

• 3330d8b9 - Allow to force to a single GPU

Compare with previous version

added 1 commit

• e1355b7f - Nits

Compare with previous version

added 1 commit

• 09b398cc - [Skip CI] Check if GPUs is available

Compare with previous version

added 1 commit

• ca65cea3 - Allow to force to a single GPU

Compare with previous version

resolved all threads

requested review from @yschneider

added 1 commit

• 837c283f - Fix lint

Compare with previous version

added 5 commits

• 837c283f...762c21ce - 2 commits from branch main
• 69c8df89 - Allow specifying device to use when training
• c1ca208f - Allow to force to a single GPU
• 09c6fb94 - Fix lint

Compare with previous version

mentioned in merge request !301 (merged)

approved this merge request

added 1 commit

• 35cea131 - Apply suggestion

Compare with previous version

resolved all threads

merged