runtime error

Exit code: 1. Reason: ine 89, in run _build_ext.run(self) File "/home/user/.pyenv/versions/3.10.13/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 345, in run self.build_extensions() File "/home/user/.pyenv/versions/3.10.13/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 525, in build_extensions _check_cuda_version(compiler_name, compiler_version) File "/home/user/.pyenv/versions/3.10.13/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 413, in _check_cuda_version raise RuntimeError(CUDA_MISMATCH_MESSAGE.format(cuda_str_version, torch.version.cuda)) RuntimeError: The detected CUDA version (11.8) mismatches the version that was used to compile PyTorch (12.1). Please make sure to use the same CUDA versions. [end of output] note: This error originates from a subprocess, and is likely not a problem with pip. error: legacy-install-failure × Encountered error while trying to install package. ╰─> flash-attn note: This is an issue with the package mentioned above, not pip. hint: See above for output from the failure. [notice] A new release of pip available: 22.3.1 -> 25.0 [notice] To update, run: pip install --upgrade pip 2025-01-30 23:13:03 | ERROR | stderr | Traceback (most recent call last): 2025-01-30 23:13:03 | ERROR | stderr | File "/home/user/app/app.py", line 10, in <module> 2025-01-30 23:13:03 | ERROR | stderr | subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U']) 2025-01-30 23:13:03 | ERROR | stderr | File "/home/user/.pyenv/versions/3.10.13/lib/python3.10/subprocess.py", line 369, in check_call 2025-01-30 23:13:03 | ERROR | stderr | raise CalledProcessError(retcode, cmd) 2025-01-30 23:13:03 | ERROR | stderr | subprocess.CalledProcessError: Command '['/home/user/.pyenv/versions/3.10.13/bin/python', '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U']' returned non-zero exit status 1.

Container logs: