nn.DataParallel использует только один графический процессор в моем коде pytorch - PullRequest
0 голосов
/ 12 апреля 2019

Приведенный ниже учебник работает хорошо, и я наблюдал те же экспериментальные результаты, то есть работают несколько графических процессоров. Так что моя экспериментальная среда не ошибается. https://github.com/pytorch/tutorials/blob/master/beginner_source/blitz/data_parallel_tutorial.py

Но, мои коды ниже показывают только один графический процессор, где строки 19, 106, 230, 241 тесно связаны с несколькими графическими процессорами.

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

Пожалуйста, помогите мне. Большое спасибо.

...