runtime error

Exit code: 1. Reason: sors: 16%|█▋ | 741M/4.55G [00:07<00:30, 126MB/s]  model.safetensors: 24%|██▎ | 1.08G/4.55G [00:08<00:19, 181MB/s] model.safetensors: 35%|███▌ | 1.61G/4.55G [00:09<00:10, 278MB/s] model.safetensors: 49%|████▉ | 2.22G/4.55G [00:10<00:06, 363MB/s] model.safetensors: 72%|███████▏ | 3.29G/4.55G [00:11<00:02, 555MB/s] model.safetensors: 86%|████████▌ | 3.89G/4.55G [00:12<00:01, 554MB/s] model.safetensors: 100%|██████████| 4.55G/4.55G [00:13<00:00, 347MB/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: