小坑:Pytorch在windows下编程时dataloader速度过慢的问题

Pytorch版本:1.7.0
在windows下编程时,num_workers > 0的话速度会非常慢。
解决办法目前还没有,只能把num_workers设为0。

在多系统的跨平台操作时,可以使用如下代码:

import os
if os.name == 'nt':
    args.num_workers = 0

Ref:
https://github.com/pytorch/pytorch/issues/12831

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容