Python MKL_THREADING_LAYER=INTEL与libgomp.so.1库不兼容

Question

!mkdir -p '{FRAME_OUTPUT_DIR}'
%cd /home/jovyan/project/DAIN

!MKL_THREADING_LAYER=AMD

!python -W ignore colab_interpolate.py --netName DAIN_slowmotion --time_step {fps/TARGET_FPS} --start_frame 1 --end_frame {frame_count} --frame_input_dir '{FRAME_INPUT_DIR}' --frame_output_dir '{FRAME_OUTPUT_DIR}'

输出这个...

我试着将FORCE设置为1，但没有任何反应，然后输出就消失了…

/home/jovyan/project/DAIN
Error: mkl-service + Intel(R) MKL: MKL_THREADING_LAYER=INTEL is incompatible with libgomp.so.1 library.
    Try to import numpy first or set the threading layer accordingly. Set MKL_SERVICE_FORCE_INTEL to force it.```

Hey there, can anyone help me solve this problem with DAIN? I'm using Saturn Cloud with pytorch but it outputs this...

Answer 1

这对我很有效。我在!python ...命令上添加了以下两行。

import os
os.environ['MKL_THREADING_LAYER'] = 'GNU'

来源：https://github.com/pytorch/pytorch/issues/37377#issuecomment-677851112