runtime error

│ │ 283 │ │ │ │ attention_mask = None │ │ 284 │ │ │ │ │ 285 │ │ │ prompt_embeds = self.text_encoder( │ │ ❱ 286 │ │ │ │ text_input_ids.to(device), │ │ 287 │ │ │ │ attention_mask=attention_mask, │ │ 288 │ │ │ ) │ │ 289 │ │ │ prompt_embeds = prompt_embeds[0] │ │ │ │ /usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py:247 in │ │ _lazy_init │ │ │ │ 244 │ │ # are found or any other error occurs │ │ 245 │ │ if 'CUDA_MODULE_LOADING' not in os.environ: │ │ 246 │ │ │ os.environ['CUDA_MODULE_LOADING'] = 'LAZY' │ │ ❱ 247 │ │ torch._C._cuda_init() │ │ 248 │ │ # Some of the queued calls may reentrantly call _lazy_init(); │ │ 249 │ │ # we need to just return without initializing in that case. │ │ 250 │ │ # However, we must not let any *other* threads in! │ ╰──────────────────────────────────────────────────────────────────────────────╯ 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: