ImportError in running the experiment

[MXC66ai]

when I run code according to README.md: cd MobileUNETR cd experiments/isic_2016/exp_2_dice_b8_a2/

the default gpu device is set to cuda:0 (you can change it)

CUDA_VISIBLE_DEVICES="0" accelerate launch run_experiment.py from .launch importFile "/usr/local/lib/python3.10/dist-packages/accelerate/utils/launch.py", line 33, in module>from ..utils.other import is_port_in_use, merge_dictsFile "/usr/local/lib/python3.10/dist-packages/accelerate/utils/other.py", line 27, in from .transformer engine import convert modelFile "/usr/local/lib/python3.10/dist-packages/accelerate/utils/transformer_engine.py", 1 ine 21, in import transformer_engine.pytorch as te File "/usr/local/lib/python3.10/dist-packages/transformer_engine/pytorch/init.py", 1 ine 6, in from .module import LayerNormLinear File "/usr/local/lib/python3.10/dist-packages/transformer_engine/pytorch/module/_init .py", line 6, in from .layernorm_linear import LayerNormLinear File "/usr/local/lib/python3.10/dist-packages/transformer_engine/pytorch/module/layernor m_linear.py", line 13, in from .. import cpp_extensions as tex File "/usr/local/lib/python3.10/dist-packages/transformer_engine/pytorch/cppextensions/ init.py", line 6, in from transformer_engine_extensions import *ImportError: /usr/local/lib/python3.10/dist-packages/transformer_engine_extensions.cpythonl-310-x86_64-linux-gnu.so: undefined symbol: _ZN5torch3jit17parseSchemaOrNameERKNSt7_cxx1112basic_stringIcSt11char_traitsIcESaIcEEE Does anyone know what this symbol is?

[srperera]

I am unsure about this particular issue. can you explain a little more about when you encounter this problem?

[MXC66ai]

The ImportError seems to be caused by an inability to open a dependent file. step1: pip install -r requirements.txt step2: cd /MobileUNETR/experiments_medical/isic_2016/exp_2_dice_b8_a2 CUDA_VISIBLE_DEVICES="0" accelerate launch run_experiment.py