runtime error

Exit code: 1. Reason: adata_for_build_wheel self.run_setup() File "/usr/local/lib/python3.10/site-packages/setuptools/build_meta.py", line 512, in run_setup super().run_setup(setup_script=setup_script) File "/usr/local/lib/python3.10/site-packages/setuptools/build_meta.py", line 317, in run_setup exec(code, locals()) File "<string>", line 227, in <module> File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1337, in CUDAExtension library_dirs += library_paths(device_type="cuda") File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1548, 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 2982, in _join_cuda_home raise OSError('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. ╰─> flash-attn note: This is an issue with the package mentioned above, not pip. hint: See above for details. Traceback (most recent call last): File "/app/app.py", line 48, in <module> sh("pip install flash-attn --no-build-isolation") File "/app/app.py", line 42, in sh def sh(cmd): subprocess.check_call(cmd, shell=True) File "/usr/local/lib/python3.10/subprocess.py", line 369, in check_call raise CalledProcessError(retcode, cmd) subprocess.CalledProcessError: Command 'pip install flash-attn --no-build-isolation' returned non-zero exit status 1. During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/app/app.py", line 74, in <module> raise gr.Error("sam2 installation failed") gradio.exceptions.Error: 'sam2 installation failed'

Container logs: