程序只能num_workers=0, 设置>0 的线程数则报错,想提高gpu利用率,达到快速训练:
参考解决方法:link
windows10下pytorch的GPU利用率低,占用率低 - 代码先锋网
需要注意的是,主程序 添加 if __name__ == '__main__':
可以提高gpu利用率。
程序只能num_workers=0, 设置>0 的线程数则报错,想提高gpu利用率,达到快速训练:
参考解决方法:link
windows10下pytorch的GPU利用率低,占用率低 - 代码先锋网
需要注意的是,主程序 添加 if __name__ == '__main__':
可以提高gpu利用率。