Pytorch：W ParallelNative.cpp:206

Question

我正在尝试按照此处的教程在我的图像集上使用预先训练的模板：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)

预先感谢您的帮助，

Answer 1

我也有同样的问题。苹果电脑。Python 3.6（也在 3.8 上重现）。pytorch 1.7。

似乎由于这个错误，数据加载器不（或不能）使用并行计算。您可以通过两种方式消除错误（这不会解决问题）。

1. 如果您可以访问数据加载器，请num_workers=0在创建数据加载器时进行设置
2. 设置环境变量 export OMP_NUM_THREADS=1

同样，这两种解决方案都会扼杀并行计算，并可能减慢数据加载（以及训练）速度。我期待 Pytorch 1.7 中的有效解决方案或补丁