环境:pytorch2.0.1_py38,torch_npu-2.0.1rc1-cp38-cp38-linux_aarch64.whl,安装了APEX混合精度模块
python3 -c "import torch;import torch_npu;print(torch_npu.npu.is_available())已经显示ture
Traceback (most recent call last): File "/home/ic314/miniconda3/envs/pytorch2.0.1_py38/lib/python3.8/site-packages/torch_npu/init.py", line 32, in <module> import torch_npu.npu File "/home/ic314/miniconda3/envs/pytorch2.0.1_py38/lib/python3.8/site-packages/torch_npu/npu/init.py", line 44, in <module> from .utils import (is_initialized, lazy_call, _lazy_init, init, set_dump, File "/home/ic314/miniconda3/envs/pytorch2.0.1_py38/lib/python3.8/site-packages/torch_npu/npu/utils.py", line 27, in <module> import torch_npu.C ImportError: /home/ic314/miniconda3/envs/pytorch2.0.1_py38/bin/../lib/libgomp.so.1: cannot allocate memory in static TLS block Segmentation fault (core dumped)
环境:pytorch2.0.1_py38,torch_npu-2.0.1rc1-cp38-cp38-linux_aarch64.whl,安装了APEX混合精度模块
python3 -c "import torch;import torch_npu;print(torch_npu.npu.is_available())已经显示ture
Traceback (most recent call last): File "/home/ic314/miniconda3/envs/pytorch2.0.1_py38/lib/python3.8/site-packages/torch_npu/init.py", line 32, in <module> import torch_npu.npu File "/home/ic314/miniconda3/envs/pytorch2.0.1_py38/lib/python3.8/site-packages/torch_npu/npu/init.py", line 44, in <module> from .utils import (is_initialized, lazy_call, _lazy_init, init, set_dump, File "/home/ic314/miniconda3/envs/pytorch2.0.1_py38/lib/python3.8/site-packages/torch_npu/npu/utils.py", line 27, in <module> import torch_npu.C ImportError: /home/ic314/miniconda3/envs/pytorch2.0.1_py38/bin/../lib/libgomp.so.1: cannot allocate memory in static TLS block Segmentation fault (core dumped)