runtime error

Exit code: 1. Reason: sr/local/lib/python3.9/importlib/__init__.py", line 127, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "<frozen importlib._bootstrap>", line 1030, in _gcd_import File "<frozen importlib._bootstrap>", line 1007, in _find_and_load File "<frozen importlib._bootstrap>", line 986, in _find_and_load_unlocked File "<frozen importlib._bootstrap>", line 680, in _load_unlocked File "<frozen importlib._bootstrap_external>", line 850, in exec_module File "<frozen importlib._bootstrap>", line 228, in _call_with_frames_removed File "/app/api.py", line 28, in <module> best_tft = TemporalFusionTransformer.load_from_checkpoint(best_model_path).to('cpu') File "/home/user/.local/lib/python3.9/site-packages/lightning/pytorch/utilities/model_helpers.py", line 125, in wrapper return self.method(cls, *args, **kwargs) File "/home/user/.local/lib/python3.9/site-packages/lightning/pytorch/core/module.py", line 1582, in load_from_checkpoint loaded = _load_from_checkpoint( File "/home/user/.local/lib/python3.9/site-packages/lightning/pytorch/core/saving.py", line 99, in _load_from_checkpoint return model.to(device) File "/home/user/.local/lib/python3.9/site-packages/lightning/fabric/utilities/device_dtype_mixin.py", line 55, in to return super().to(*args, **kwargs) File "/home/user/.local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1340, in to return self._apply(convert) File "/home/user/.local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/home/user/.local/lib/python3.9/site-packages/torchmetrics/metric.py", line 908, in _apply _dummy_tensor = fn(torch.zeros(1, device=self.device)) File "/home/user/.local/lib/python3.9/site-packages/torch/cuda/__init__.py", line 319, 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: