runtime error

Exit code: 1. Reason: Output: Errors: Traceback (most recent call last): File "/home/user/app/setup.py", line 70, in <module> ext_modules=get_extensions(), File "/home/user/app/setup.py", line 51, in get_extensions ext_modules = [CUDAExtension("sam2._C", srcs, extra_compile_args=compile_args)] File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1279, in CUDAExtension library_dirs += library_paths(device_type="cuda") File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1490, 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 2889, 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. Command failed with return code: 1 Traceback (most recent call last): File "/home/user/app/app_vid.py", line 171, in <module> if torch.cuda.get_device_properties(0).major >= 8: File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 576, in get_device_properties _lazy_init() # will define _get_device_properties File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 372, in _lazy_init torch._C._cuda_init() RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

Container logs: