win10下 pytorch 跑模型 gpu利用率低

查阅资料后发现 Dataloader中的num_workers参数(线程数)设置为0,该为4后,nvidia-smi查看GPU占用率变为高(不要用任务管理器查看)

posted @ 2022-07-28 16:28  lcxxx  Views(171)  Comments(0Edit  收藏  举报