runtime error

4M [00:01<00:33, 16.1MB/s] model-00002-of-00002.safetensors: 85%|████████▌ | 480M/564M [00:02<00:00, 245MB/s]  model-00002-of-00002.safetensors: 100%|█████████▉| 564M/564M [00:02<00:00, 235MB/s] Downloading shards: 100%|██████████| 2/2 [00:28<00:00, 12.37s/it] Downloading shards: 100%|██████████| 2/2 [00:28<00:00, 14.46s/it] Traceback (most recent call last): File "/home/user/app/app.py", line 6, in <module> model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype="auto", trust_remote_code=True) File "/home/user/.local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 561, in from_pretrained return model_class.from_pretrained( File "/home/user/.local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3594, in from_pretrained model = cls(config, *model_args, **model_kwargs) File "/home/user/.cache/huggingface/modules/transformers_modules/microsoft/phi-2/3f879ff35d20d910f0968d0b9b35a9fc074ecb27/modeling_phi.py", line 970, in __init__ self.model = PhiModel(config) File "/home/user/.cache/huggingface/modules/transformers_modules/microsoft/phi-2/3f879ff35d20d910f0968d0b9b35a9fc074ecb27/modeling_phi.py", line 821, in __init__ self.embed_tokens = nn.Embedding(config.vocab_size, config.hidden_size, self.padding_idx) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/sparse.py", line 143, in __init__ self.weight = Parameter(torch.empty((num_embeddings, embedding_dim), **factory_kwargs), File "/home/user/.local/lib/python3.10/site-packages/torch/utils/_device.py", line 77, in __torch_function__ return func(*args, **kwargs) File "/home/user/.local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 302, 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: