runtime error

Exit code: 1. Reason: sion import DPMS, FlowEuler File "/home/user/app/diffusion/__init__.py", line 6, in <module> from .dpm_solver import DPMS File "/home/user/app/diffusion/dpm_solver.py", line 20, in <module> from .model.dpm_solver import DPM_Solver, NoiseScheduleFlow, NoiseScheduleVP, model_wrapper File "/home/user/app/diffusion/model/dpm_solver.py", line 23, in <module> from .nets.sana_blocks import ( File "/home/user/app/diffusion/model/nets/__init__.py", line 1, in <module> from .sana import ( File "/home/user/app/diffusion/model/nets/sana.py", line 46, in <module> from diffusion.model.nets.fastlinear.modules import TritonLiteMLA, TritonMBConvPreGLU File "/home/user/app/diffusion/model/nets/fastlinear/modules/__init__.py", line 17, in <module> from .triton_lite_mla import * File "/home/user/app/diffusion/model/nets/fastlinear/modules/triton_lite_mla.py", line 24, in <module> from .triton_lite_mla_kernels.linear_relu_fwd import linear_relu_fwd File "/home/user/app/diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/linear_relu_fwd.py", line 99, in <module> def linear_relu_fwd_kernel( File "/home/user/app/diffusion/model/nets/fastlinear/modules/utils/custom_autotune.py", line 108, in decorator return CustomAutotuner( File "/home/user/app/diffusion/model/nets/fastlinear/modules/utils/custom_autotune.py", line 37, in __init__ torch.cuda.get_device_name(0).replace(" ", "_"), File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 491, in get_device_name return get_device_properties(device).name File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 523, in get_device_properties _lazy_init() # will define _get_device_properties File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 319, 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: