runtime error

Exit code: 1. Reason: ��▋ | 741M/4.55G [00:07<00:26, 141MB/s]  model.safetensors: 38%|███▊ | 1.75G/4.55G [00:08<00:07, 359MB/s] model.safetensors: 50%|█████ | 2.28G/4.55G [00:09<00:05, 406MB/s] model.safetensors: 62%|██████▏ | 2.80G/4.55G [00:11<00:04, 424MB/s] model.safetensors: 82%|████████▏ | 3.74G/4.55G [00:12<00:01, 523MB/s] model.safetensors: 97%|█████████▋| 4.41G/4.55G [00:13<00:00, 561MB/s] model.safetensors: 100%|██████████| 4.55G/4.55G [00:13<00:00, 339MB/s] Number of parameters in VAE: 340.30M Loaded VAE from /home/user/.cache/huggingface/hub/models--nvidia--PartPacker/snapshots/17046fe1729d05db99cb11d7cc6885d777fcfc73/vae.pt Number of parameters in DiT: 1249.51M Traceback (most recent call last): File "/home/user/app/app.py", line 58, in <module> model = Model(model_config).eval().cuda().bfloat16() File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1050, in cuda return self._apply(lambda t: t.cuda(device)) 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 1050, in <lambda> return self._apply(lambda t: t.cuda(device)) 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: