runtime error

Exit code: 1. Reason: ckages (from torch->flash-attn) (3.2.0) Requirement already satisfied: sympy==1.13.1 in /usr/local/lib/python3.10/site-packages (from torch->flash-attn) (1.13.1) Requirement already satisfied: mpmath<1.4,>=1.1.0 in /usr/local/lib/python3.10/site-packages (from sympy==1.13.1->torch->flash-attn) (1.3.0) Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/site-packages (from jinja2->torch->flash-attn) (2.1.5) Downloading einops-0.8.1-py3-none-any.whl (64 kB) Building wheels for collected packages: flash-attn Building wheel for flash-attn (setup.py): started Building wheel for flash-attn (setup.py): finished with status 'done' Created wheel for flash-attn: filename=flash_attn-2.8.2-py3-none-any.whl size=255921868 sha256=b57b269107bc36e934ade6fff1fc4682bcd8174eabbfea10f84bdfd295e6df0b Stored in directory: /home/user/.cache/pip/wheels/ae/2e/e8/9e6fe4527684a326f08d81f12715d78df48f568cb052190033 Successfully built flash-attn Installing collected packages: einops, flash-attn Successfully installed einops-0.8.1 flash-attn-2.8.2 [notice] A new release of pip is available: 25.0.1 -> 25.2 [notice] To update, run: /usr/local/bin/python -m pip install --upgrade pip Traceback (most recent call last): File "/home/user/app/app.py", line 2, in <module> import flash_attn ModuleNotFoundError: No module named 'flash_attn' During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/app.py", line 12, in <module> import flash_attn File "/usr/local/lib/python3.10/site-packages/flash_attn/__init__.py", line 3, in <module> from flash_attn.flash_attn_interface import ( File "/usr/local/lib/python3.10/site-packages/flash_attn/flash_attn_interface.py", line 15, in <module> import flash_attn_2_cuda as flash_attn_gpu ImportError: /usr/local/lib/python3.10/site-packages/flash_attn_2_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c105ErrorC2ENS_14SourceLocationENSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE

Container logs: