Pytorch1.5.1+cuda9.2训练mnist提高gpu利用率
·
程序只能num_workers=0, 设置>0 的线程数则报错，想提高gpu利用率，达到快速训练：
参考解决方法：link
windows10下pytorch的GPU利用率低，占用率低 - 代码先锋网
需要注意的是，主程序 添加 if __name__ == '__main__':
可以提高gpu利用率。
更多推荐
0
0- 0
已为社区贡献1条内容
已为社区贡献1条内容
所有评论(0)