runtime error

Exit code: 1. Reason: │ │ 902 │ │ Returns: │ │ 903 │ │ │ Module: self │ │ 904 │ │ """ │ │ ❱ 905 │ │ return self._apply(lambda t: t.cuda(device)) │ │ 906 │ │ │ 907 │ def ipu(self: T, device: Optional[Union[int, device]] = None) -> │ │ 908 │ │ r"""Moves all model parameters and buffers to the IPU. │ │ │ │ /usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py:247 in │ │ _lazy_init │ │ │ │ 244 │ │ # are found or any other error occurs │ │ 245 │ │ if 'CUDA_MODULE_LOADING' not in os.environ: │ │ 246 │ │ │ os.environ['CUDA_MODULE_LOADING'] = 'LAZY' │ │ ❱ 247 │ │ torch._C._cuda_init() │ │ 248 │ │ # Some of the queued calls may reentrantly call _lazy_init(); │ │ 249 │ │ # we need to just return without initializing in that case. │ │ 250 │ │ # 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: