runtime error

Exit code: 1. Reason: 6.57G/7.18G [00:24<00:01, 440MB/s] stage_c_bf16.safetensors: 100%|██████████| 7.18G/7.18G [00:25<00:00, 477MB/s] stage_c_bf16.safetensors: 100%|██████████| 7.18G/7.18G [00:25<00:00, 285MB/s] ultrapixel_t2i.safetensors: 0%| | 0.00/211M [00:00<?, ?B/s] ultrapixel_t2i.safetensors: 5%|▍ | 10.0M/211M [00:01<00:22, 8.98MB/s] ultrapixel_t2i.safetensors: 100%|██████████| 211M/211M [00:01<00:00, 109MB/s] ['model_version', 'effnet_checkpoint_path', 'previewer_checkpoint_path', 'use_ddp'] Running in single thread, DDP not enabled. ['model_version', 'stage_a_checkpoint_path', 'effnet_checkpoint_path'] ['transforms', 'clip_preprocess', 'gdf', 'sampling_configs', 'effnet_preprocess'] Traceback (most recent call last): File "/home/user/app/app.py", line 526, in <module> models = core.setup_models(extras) File "/home/user/app/train/train_t2i.py", line 217, in setup_models effnet.eval().requires_grad_(False).to(self.device) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1369, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 928, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 928, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 928, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 955, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1355, in convert return t.to( File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 412, 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: