runtime error

[INFO] loading options.. Traceback (most recent call last): File "/home/user/app/raymarching/raymarching.py", line 10, in <module> import _raymarching as _backend ModuleNotFoundError: No module named '_raymarching' During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/app.py", line 84, in <module> from nerf.network_grid import NeRFNetwork File "/home/user/app/nerf/network_grid.py", line 6, in <module> from .renderer import NeRFRenderer File "/home/user/app/nerf/renderer.py", line 12, in <module> import raymarching File "/home/user/app/raymarching/__init__.py", line 1, in <module> from .raymarching import * File "/home/user/app/raymarching/raymarching.py", line 12, in <module> from .backend import _backend File "/home/user/app/raymarching/backend.py", line 31, in <module> _backend = load(name='_raymarching', File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1284, in load return _jit_compile( File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1509, in _jit_compile _write_ninja_file_and_build_library( File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1601, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1699, in _prepare_ldflags extra_ldflags.append(f'-L{_join_cuda_home("lib64")}') File "/home/user/.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.

Container logs: