Pytorch:W ParallelNative.cpp:206
Posted
技术标签:
【中文标题】Pytorch:W ParallelNative.cpp:206【英文标题】:Pytorch : W ParallelNative.cpp:206 【发布时间】:2021-02-22 14:15:16 【问题描述】:我正在尝试按照此处的教程在我的图像集上使用预先训练的模板: https://pytorch.org/tutorials/beginner/finetuning_torchvision_models_tutorial.html
只有当我运行我的代码并且控制台锁定时,我总是得到这个“错误”:
[W ParallelNative.cpp:206] Warning: Cannot set number of intraop threads after parallel work has started or after set_num_threads call when using native parallel backend (function set_num_threads)
提前感谢您的帮助,
【问题讨论】:
请更新整个错误消息和文本格式。 【参考方案1】:我也有同样的问题。 苹果。 Python 3.6(也在 3.8 上重现)。 Pytorch 1.7。
似乎有这个错误数据加载器不(或不能)使用并行计算。 您可以通过两种方式消除错误(这不会解决问题)。
-
如果您可以访问您的数据加载器,请在创建数据加载器时设置
num_workers=0
设置环境变量export OMP_NUM_THREADS=1
同样,这两种解决方案都会终止并行计算,并可能会减慢数据加载速度(从而减慢训练速度)。我期待 Pytorch 1.7 中的高效解决方案或补丁
【讨论】:
我也没有,正在寻找另一种解决方法以上是关于Pytorch:W ParallelNative.cpp:206的主要内容,如果未能解决你的问题,请参考以下文章
小白学Pytorch系列--Torch.nn API Shuffle Layers(16)