runtime error

Exit code: 1. Reason: it [00:00, ?it/s] 0it [00:00, ?it/s] Traceback (most recent call last): File "/home/user/app/app.py", line 46, in <module> model = SF3D.from_pretrained( File "/home/user/app/sf3d/system.py", line 89, in from_pretrained model = cls(cfg) File "/home/user/app/sf3d/models/utils.py", line 29, in __init__ self.configure(*args, **kwargs) File "/home/user/app/sf3d/system.py", line 139, in configure self.baker = TextureBaker() File "/home/user/app/sf3d/texture_baker.py", line 13, in __init__ self.baker = slangtorch.loadModule( File "/usr/local/lib/python3.10/site-packages/slangtorch/slangtorch.py", line 617, in loadModule rawModule = _loadModule(fileName, moduleName, buildDir, options, sourceDir=outputFolder, verbose=verbose, includePaths=includePaths, dryRun=False) File "/usr/local/lib/python3.10/site-packages/slangtorch/slangtorch.py", line 536, in _loadModule slangLib, metadata = compileAndLoadModule( File "/usr/local/lib/python3.10/site-packages/slangtorch/slangtorch.py", line 427, in compileAndLoadModule slangLib = _compileAndLoadModule(metadata, sources, moduleName, buildDir, slangSourceDir, verbose) File "/usr/local/lib/python3.10/site-packages/slangtorch/slangtorch.py", line 466, in _compileAndLoadModule return jit_compile( File "/usr/local/lib/python3.10/site-packages/slangtorch/util/compile.py", line 71, in jit_compile _write_ninja_file_and_build_library( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1800, 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 1899, 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 2416, 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: