How to specify gpu like pytorch.device('cuda:2') in fastai?

fastai fastai dev
2

If you use torch.cuda.set_device(2) everything should be on that GPU.

2 Likes