nn.DataParallel在我的pytorch代码中仅使用一个GPU

时间:2019-04-12 12:26:08

标签: deep-learning gpu pytorch

以下教程效果很好,而且我观察到了相同的实验结果,即正在运行多个GPU。所以我的实验环境没有错。 https://github.com/pytorch/tutorials/blob/master/beginner_source/blitz/data_parallel_tutorial.py

但是,我在下面的代码中只吃了一个GPU,其中第19、106、230、241行与多GPU紧密相关。

https://github.com/convexsetgithub/mytextgan/blob/master/train_gpuS.py

请帮助我。非常感谢。

0 个答案:

没有答案