runtime error

Exit code: 1. Reason: s/transformers/deepspeed.py:23: FutureWarning: transformers.deepspeed module is deprecated and will be removed in a future version. Please import deepspeed modules directly from transformers.integrations warnings.warn( [2024-12-13 09:45:31,023] [INFO] [real_accelerator.py:203:get_accelerator] Setting ds_accelerator to cuda (auto detect) df: /home/user/.triton/autotune: No such file or directory Traceback (most recent call last): File "/home/user/app/app.py", line 20, in <module> from minigemini.constants import DEFAULT_IMAGE_TOKEN, IMAGE_TOKEN_INDEX File "/home/user/app/minigemini/__init__.py", line 3, in <module> from .model import MiniGeminiLlamaForCausalLM File "/home/user/app/minigemini/model/__init__.py", line 1, in <module> from .language_model.mini_gemini_llama import MiniGeminiLlamaForCausalLM File "/home/user/app/minigemini/model/language_model/mini_gemini_llama.py", line 31, in <module> from minigemini.model.mini_gemini_arch import MiniGeminiMetaModel, MiniGeminiMetaForCausalLM File "/home/user/app/minigemini/model/mini_gemini_arch.py", line 29, in <module> import deepspeed File "/usr/local/lib/python3.10/site-packages/deepspeed/__init__.py", line 25, in <module> from . import ops File "/usr/local/lib/python3.10/site-packages/deepspeed/ops/__init__.py", line 15, in <module> from ..git_version_info import compatible_ops as __compatible_ops__ File "/usr/local/lib/python3.10/site-packages/deepspeed/git_version_info.py", line 29, in <module> op_compatible = builder.is_compatible() File "/usr/local/lib/python3.10/site-packages/deepspeed/ops/op_builder/fp_quantizer.py", line 35, in is_compatible sys_cuda_major, _ = installed_cuda_version() File "/usr/local/lib/python3.10/site-packages/deepspeed/ops/op_builder/builder.py", line 51, in installed_cuda_version raise MissingCUDAException("CUDA_HOME does not exist, unable to compile CUDA op(s)") deepspeed.ops.op_builder.builder.MissingCUDAException: CUDA_HOME does not exist, unable to compile CUDA op(s)

Container logs: