runtime error

Exit code: 1. Reason: fetensors: 90%|█████████ | 22.9G/25.4G [01:09<00:07, 308MB/s] consolidated.safetensors: 92%|█████████▏| 23.3G/25.4G [01:11<00:07, 277MB/s] consolidated.safetensors: 93%|█████████▎| 23.6G/25.4G [01:12<00:06, 268MB/s] consolidated.safetensors: 95%|█████████▍| 24.0G/25.4G [01:13<00:04, 312MB/s] consolidated.safetensors: 97%|█████████▋| 24.6G/25.4G [01:14<00:01, 380MB/s] consolidated.safetensors: 99%|█████████▉| 25.2G/25.4G [01:15<00:00, 423MB/s] consolidated.safetensors: 100%|██████████| 25.4G/25.4G [01:15<00:00, 335MB/s] params.json: 0%| | 0.00/453 [00:00<?, ?B/s] params.json: 100%|██████████| 453/453 [00:00<00:00, 3.83MB/s] tekken.json: 0%| | 0.00/19.3M [00:00<?, ?B/s] tekken.json: 100%|██████████| 19.3M/19.3M [00:00<00:00, 70.3MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 62, in <module> model = Transformer.from_folder(model_dir) File "/usr/local/lib/python3.10/site-packages/mistral_inference/transformer.py", line 338, in from_folder return model.to(device=device, dtype=dtype) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1355, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 915, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 942, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1341, in convert return t.to( 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: