runtime error

Exit code: 1. Reason: ��██ | 1.40G/4.55G [00:07<00:14, 222MB/s] model.safetensors: 37%|███▋ | 1.66G/4.55G [00:09<00:15, 185MB/s] model.safetensors: 48%|████▊ | 2.20G/4.55G [00:10<00:08, 268MB/s] model.safetensors: 63%|██████▎ | 2.87G/4.55G [00:11<00:04, 368MB/s] model.safetensors: 75%|███████▍ | 3.41G/4.55G [00:12<00:02, 398MB/s] model.safetensors: 93%|█████████▎| 4.21G/4.55G [00:13<00:00, 493MB/s] model.safetensors: 100%|██████████| 4.55G/4.55G [00:14<00:00, 317MB/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 57, 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: