runtime error

Exit code: 1. Reason: > load( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1380, in load return _jit_compile( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1798, in _jit_compile _write_ninja_file_and_build_library( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1895, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1992, in _prepare_ldflags if (not os.path.exists(_join_cuda_home(extra_lib_dir)) and File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2525, 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. During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/app.py", line 7, in <module> from rwkv.model import RWKV File "/usr/local/lib/python3.10/site-packages/rwkv/model.py", line 42, in <module> load( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1380, in load return _jit_compile( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1798, in _jit_compile _write_ninja_file_and_build_library( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1895, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1992, in _prepare_ldflags if (not os.path.exists(_join_cuda_home(extra_lib_dir)) and File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2525, 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.

Container logs: