runtime error

1, in <module> CUDAExtension( File "/home/user/.local/lib/python3.9/site-packages/torch/utils/cpp_extension.py", line 1048, in CUDAExtension library_dirs += library_paths(cuda=True) File "/home/user/.local/lib/python3.9/site-packages/torch/utils/cpp_extension.py", line 1179, in library_paths if (not os.path.exists(_join_cuda_home(lib_dir)) and File "/home/user/.local/lib/python3.9/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. Traceback (most recent call last): File "/home/user/app/app.py", line 6, in <module> from train import * File "/home/user/app/train.py", line 19, in <module> from gaussian_renderer import render, network_gui File "/home/user/app/gaussian_renderer/__init__.py", line 14, in <module> from diff_gaussian_rasterization import GaussianRasterizationSettings, GaussianRasterizer ModuleNotFoundError: No module named 'diff_gaussian_rasterization' During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/app.py", line 17, in <module> from train import * File "/home/user/app/train.py", line 19, in <module> from gaussian_renderer import render, network_gui File "/home/user/app/gaussian_renderer/__init__.py", line 14, in <module> ==> diff-gaussian-rasterization installed! from diff_gaussian_rasterization import GaussianRasterizationSettings, GaussianRasterizer ModuleNotFoundError: No module named 'diff_gaussian_rasterization'

Container logs: