search

HumanCompatibleAI / imitation

Clean PyTorch implementations of imitation and reward learning algorithms
https://imitation.readthedocs.io/
MIT License
1.3k stars 248 forks source link

abort and kernel crash with OMP libomp.dylib #844

Closed chenyangkang closed 6 months ago

chenyangkang commented 6 months ago

Bug description

while importing:

from imitation.policies.serialize import load_policy

crashed with:

OMP: Error #15: Initializing libomp.dylib, but found libomp.dylib 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://openmp.llvm.org/
zsh: abort      python

Environment

chenyangkang commented 6 months ago

This solved the problem:

import os
os.environ['KMP_DUPLICATE_LIB_OK']='True'