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 反向梯度计算问题

小白学Pytorch系列--Torch.nn API Shuffle Layers(16)

pytorch学习笔记:线性模型

深度之眼PyTorch训练营第二期 ---3计算图与动态图机制

AFM模型 pytorch示例代码

AFM模型 pytorch示例代码