runtime error

Exit code: 1. Reason: /usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 158, 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) The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 38, in <module> from vall_e.inference import TTS, cfg File "/usr/local/lib/python3.10/site-packages/vall_e/inference.py", line 18, in <module> from .emb.transcribe import transcribe File "/usr/local/lib/python3.10/site-packages/vall_e/emb/transcribe.py", line 21, in <module> from transformers import pipeline File "<frozen importlib._bootstrap>", line 1075, in _handle_fromlist File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1955, in __getattr__ module = self._get_module(self._class_to_module[name]) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1969, in _get_module raise RuntimeError( RuntimeError: Failed to import transformers.pipelines because of the following error (look up to see its traceback): CUDA_HOME does not exist, unable to compile CUDA op(s)

Container logs: