windows环境下pytorch使用num_worker
# 在 pytorch 中,我们可以通过设置 num_worker 的数量来提高数据加载的速度,从而减少将数据
从 CPU 加载到 GPU 的时间开销,以提高 GPU 的利用率,进而加快模型的训练速度。
# 在 linux 环境下设置 dataLoader 的 num_worker 数量大于 0 是可以正常运行的,但是在
Windows 环境下会报错,只能设置 num_worker=0 才可以正常运行,但是这样会使得模型
的训练速度极其漫长…
# 如果还是想在 Windows 环境下在 pytorch 中启用多线程加载数据,那么应该怎么办呢?
这个问题我找了很久很久很久…...
more...