runtime error

Exit code: 1. Reason: │ │ 746 │ │ Returns: │ │ 747 │ │ │ Module: self │ │ 748 │ │ """ │ │ ❱ 749 │ │ return self._apply(lambda t: t.cuda(device)) │ │ 750 │ │ │ 751 │ def ipu(self: T, device: Optional[Union[int, device]] = None) -> │ │ 752 │ │ r"""Moves all model parameters and buffers to the IPU. │ │ │ │ /usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py:229 in │ │ _lazy_init │ │ │ │ 226 │ │ # are found or any other error occurs │ │ 227 │ │ if 'CUDA_MODULE_LOADING' not in os.environ: │ │ 228 │ │ │ os.environ['CUDA_MODULE_LOADING'] = 'LAZY' │ │ ❱ 229 │ │ torch._C._cuda_init() │ │ 230 │ │ # Some of the queued calls may reentrantly call _lazy_init(); │ │ 231 │ │ # we need to just return without initializing in that case. │ │ 232 │ │ # However, we must not let any *other* threads in! │ ╰──────────────────────────────────────────────────────────────────────────────╯ 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: