runtime error

| 8/8 [04:15<00:00, 31.92s/it] Loading checkpoint shards: 0%| | 0/8 [00:00<?, ?it/s] Loading checkpoint shards: 12%|█▎ | 1/8 [00:06<00:43, 6.27s/it] Loading checkpoint shards: 25%|██▌ | 2/8 [00:14<00:45, 7.64s/it] Loading checkpoint shards: 38%|███▊ | 3/8 [00:23<00:39, 7.90s/it] Loading checkpoint shards: 50%|█████ | 4/8 [00:32<00:34, 8.70s/it] Loading checkpoint shards: 62%|██████▎ | 5/8 [00:40<00:25, 8.36s/it] Loading checkpoint shards: 75%|███████▌ | 6/8 [00:46<00:15, 7.63s/it] Loading checkpoint shards: 88%|████████▊ | 7/8 [00:51<00:06, 6.63s/it] Loading checkpoint shards: 100%|██████████| 8/8 [00:55<00:00, 5.83s/it] Loading checkpoint shards: 100%|██████████| 8/8 [00:55<00:00, 6.96s/it] Traceback (most recent call last): File "/home/user/app/web_demo_old.py", line 6, in <module> model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda() File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 905, in cuda return self._apply(lambda t: t.cuda(device)) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 797, in _apply module._apply(fn) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 797, in _apply module._apply(fn) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 820, in _apply param_applied = fn(param) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 905, in <lambda> return self._apply(lambda t: t.cuda(device)) File "/home/user/.local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 247, 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: