Pytorch | OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.

今天用YOLOv5训练数据集,跑train.py出现以下报错:

OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program. That is dangerous, since it can degrade performance or cause incorrect results. The best thing to do is to ensure that only a single OpenMP runtime is linked into the process, e.g. by avoiding static linking of the OpenMP runtime in any library. As an unsafe, unsupported, undocumented workaround you can set the environment variable KMP_DUPLICATE_LIB_OK=TRUE to allow the program to continue to execute, but that may cause crashes or silently produce incorrect results. For more information, please see http://www.intel.com/software/products/support/.

解决如下:

在运行代码的前面添加两行代码即可,亲测有效
在这里插入图片描述

import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"

参考文章: https://www.cnblogs.com/sunshine-66/p/15920386.html

猜你喜欢

转载自blog.csdn.net/weixin_43401024/article/details/124527342
今日推荐