runtime error

Exit code: 1. Reason: llm_pretrain.pt: 15%|█▍ | 1.23G/8.27G [00:15<00:40, 173MB/s] llm_pretrain.pt: 28%|██▊ | 2.30G/8.27G [00:17<00:17, 333MB/s] llm_pretrain.pt: 41%|████ | 3.37G/8.27G [00:18<00:10, 480MB/s] llm_pretrain.pt: 50%|████▉ | 4.11G/8.27G [00:19<00:08, 487MB/s] llm_pretrain.pt: 76%|███████▌ | 6.26G/8.27G [00:20<00:02, 858MB/s] llm_pretrain.pt: 89%|████████▉ | 7.40G/8.27G [00:23<00:01, 668MB/s] llm_pretrain.pt: 100%|██████████| 8.27G/8.27G [00:23<00:00, 350MB/s] Traceback (most recent call last): File "/app/app.py", line 353, in <module> initiate_model('pretrained_models/FireRedTTS2') File "/app/app.py", line 21, in initiate_model model = FireRedTTS2( File "/app/fireredtts2/fireredtts2.py", line 39, in __init__ self._model = load_llm_model( File "/app/fireredtts2/llm/utils.py", line 257, in load_llm_model model = model.to(device=device) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1371, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 930, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 930, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 930, in _apply module._apply(fn) [Previous line repeated 2 more times] File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 957, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1357, in convert return t.to( File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 410, 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: