Pytorch1.5.1+cuda9.2训练mnist提高gpu利用率

程序只能num_workers=0, 设置>0 的线程数则报错,想提高gpu利用率,达到快速训练:

参考解决方法:link

windows10下pytorch的GPU利用率低,占用率低 - 代码先锋网

需要注意的是,主程序 添加 if __name__ == '__main__':

可以提高gpu利用率。

猜你喜欢

转载自blog.csdn.net/huachuchengzhang/article/details/122815845