runtime error

Exit code: 1. Reason: re[import] Device: cpu torch_dtype: torch.float16 /usr/local/lib/python3.10/dist-packages/torch/cuda/__init__.py:61: FutureWarning: The pynvml package is deprecated. Please install nvidia-ml-py instead. If you did not install pynvml directly, please report this to the maintainers of the package that installed pynvml for you. import pynvml # type: ignore[import] Multiple distributions found for package optimum. Picked distribution: optimum Traceback (most recent call last): File "/home/user/app/./server/main.py", line 350, in <module> pipeline_instance = pipeline_class(config, device, torch_dtype) File "/home/user/app/server/pipelines/pix2pixTurbo.py", line 103, in __init__ self.model = Pix2Pix_Turbo("edge_to_image") File "/home/user/app/server/pipelines/pix2pix/pix2pix_turbo.py", line 42, in __init__ ).cuda() File "/usr/local/lib/python3.10/dist-packages/transformers/modeling_utils.py", line 3654, in cuda return super().cuda(*args, **kwargs) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1050, in cuda return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 927, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1050, in <lambda> return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/dist-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: