runtime error

3.66G [01:32<00:00, 37.7MB/s] pytorch_model-00002-of-00002.bin: 100%|██████████| 3.66G/3.66G [01:33<00:00, 25.8MB/s] pytorch_model-00002-of-00002.bin: 100%|██████████| 3.66G/3.66G [01:33<00:00, 39.1MB/s] Downloading shards: 100%|██████████| 2/2 [05:50<00:00, 160.91s/it] Downloading shards: 100%|██████████| 2/2 [05:50<00:00, 175.29s/it] Loading checkpoint shards: 0%| | 0/2 [00:00<?, ?it/s] Loading checkpoint shards: 0%| | 0/2 [00:22<?, ?it/s] Traceback (most recent call last): File "/home/user/app/app.py", line 40, in <module> model, tokenizer = init_model() File "/home/user/app/app.py", line 11, in init_model model = AutoModelForCausalLM.from_pretrained("Linly-AI/Chinese-LLaMA-2-7B-hf", device_map="cuda:0", File "/home/user/.local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 493, in from_pretrained return model_class.from_pretrained( File "/home/user/.local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2903, in from_pretrained ) = cls._load_pretrained_model( File "/home/user/.local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3260, in _load_pretrained_model new_error_msgs, offload_index, state_dict_index = _load_state_dict_into_meta_model( File "/home/user/.local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 717, in _load_state_dict_into_meta_model set_module_tensor_to_device(model, param_name, param_device, **set_module_kwargs) File "/home/user/.local/lib/python3.10/site-packages/accelerate/utils/modeling.py", line 298, in set_module_tensor_to_device new_value = value.to(device) File "/home/user/.local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 216, 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: