Does spotlight support multi gpu training?

maciejkula / spotlight

Deep recommender models using PyTorch.

MIT License

2.99k stars 423 forks source link

Open betterpotter opened 6 years ago

betterpotter commented 6 years ago

Multi-gpu training is supported by PyTorch. Does spotlight support multi gpu training?

I tried os.environ["CUDA_VISIBLE_DEVICES"] with more than one gpus, however, it did not work.

maciejkula commented 5 years ago

I don't think so. Let me know if you find a way to do it!