Hi!
I have ulmfit classifier and several gpu. I load classifier with function load_classifier() and it always load on gpu-0. I try torch.cuda.set_device(1), but still it loads to gpu-0.
How to choose another gpu?
Hi!
I have ulmfit classifier and several gpu. I load classifier with function load_classifier() and it always load on gpu-0. I try torch.cuda.set_device(1), but still it loads to gpu-0.
How to choose another gpu?