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?