runtime error

Exit code: 1. Reason: all last): File "<string>", line 2, in <module> File "<pip-setuptools-caller>", line 34, in <module> File "/tmp/pip-install-95g0pyjw/flash-attn_ebf79ee468ff46c09e0dcba2332c3b10/setup.py", line 179, in <module> CUDAExtension( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1048, in CUDAExtension library_dirs += library_paths(cuda=True) File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1179, in library_paths if (not os.path.exists(_join_cuda_home(lib_dir)) and File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2223, in _join_cuda_home raise EnvironmentError('CUDA_HOME environment variable is not set. ' OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root. [end of output] note: This error originates from a subprocess, and is likely not a problem with pip. error: metadata-generation-failed × Encountered error while generating package metadata. ╰─> See above for output. note: This is an issue with the package mentioned above, not pip. hint: See above for details. [notice] A new release of pip available: 22.3.1 -> 24.2 [notice] To update, run: pip install --upgrade pip 2024-10-24 01:19:17 | ERROR | stderr | Traceback (most recent call last): 2024-10-24 01:19:17 | ERROR | stderr | File "/home/user/app/app.py", line 15, in <module> 2024-10-24 01:19:17 | ERROR | stderr | subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U']) 2024-10-24 01:19:17 | ERROR | stderr | File "/usr/local/lib/python3.10/subprocess.py", line 369, in check_call 2024-10-24 01:19:17 | ERROR | stderr | raise CalledProcessError(retcode, cmd) 2024-10-24 01:19:17 | ERROR | stderr | subprocess.CalledProcessError: Command '['/usr/local/bin/python', '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U']' returned non-zero exit status 1.

Container logs: