использование torch.multiprocessing на нескольких графических процессорах - PullRequest
1 голос
/ 18 июня 2020

Я пытаюсь выполнить вывод на изображениях параллельно, используя многопроцессорность torch. Странно то, что я вижу, что в nvtop используется только один графический процессор, а производительность ужасная. работа. Что также странно, оба графических процессора показывают выделенную память, но GPU 0 в два раза больше, чем GPU 1. Есть мысли о том, что у меня здесь не так? Спасибо.

...