runtime error

Exit code: 1. Reason: 00.37 examples/s] Generating full split: 40910 examples [00:06, 6174.42 examples/s] Generating full split: 47383 examples [00:07, 6151.02 examples/s] Generating full split: 53705 examples [00:08, 5943.52 examples/s] Generating full split: 59864 examples [00:09, 5899.59 examples/s] Generating full split: 66489 examples [00:10, 5706.21 examples/s] Generating full split: 72944 examples [00:11, 5843.83 examples/s] Generating full split: 79142 examples [00:12, 5941.33 examples/s] Generating full split: 85951 examples [00:13, 6158.61 examples/s] Generating full split: 92825 examples [00:15, 6328.50 examples/s] Generating full split: 99430 examples [00:16, 5963.44 examples/s] Generating full split: 106153 examples [00:17, 6164.25 examples/s] Generating full split: 112518 examples [00:18, 6164.17 examples/s] Generating full split: 112590 examples [00:18, 6143.46 examples/s] Traceback (most recent call last): File "/home/user/app/app.py", line 182, in <module> train_model(ncf_model, train_loader, test_loader, epochs=10, lr=1e-4) File "/home/user/app/app.py", line 117, in train_model model.to("cuda") # Move model to GPU File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1145, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 797, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 820, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1143, in convert return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 247, 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: