runtime error

Exit code: 1. Reason: ing data: 100%|██████████| 173k/173k [00:00<00:00, 62.3MB/s] Generating test split: 0 examples [00:00, ? examples/s] Generating test split: 1 examples [00:00, 411.61 examples/s] preprocessor_config.json: 0%| | 0.00/266 [00:00<?, ?B/s] preprocessor_config.json: 100%|██████████| 266/266 [00:00<00:00, 2.04MB/s] config.json: 0%| | 0.00/69.5k [00:00<?, ?B/s] config.json: 100%|██████████| 69.5k/69.5k [00:00<00:00, 150MB/s] model.safetensors: 0%| | 0.00/46.8M [00:00<?, ?B/s] model.safetensors: 100%|██████████| 46.8M/46.8M [00:00<00:00, 133MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 19, in <module> model = AutoModelForImageClassification.from_pretrained("microsoft/resnet-18").to("cuda") File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3164, in to return super().to(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1340, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) [Previous line repeated 1 more time] File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 927, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1326, in convert return t.to( File "/usr/local/lib/python3.10/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: