runtime error

quivalent to passing `weights=AlexNet_Weights.IMAGENET1K_V1`. You can also use `weights=AlexNet_Weights.DEFAULT` to get the most up-to-date weights. warnings.warn(msg) Downloading: "https://download.pytorch.org/models/alexnet-owt-7be5be79.pth" to /home/user/.cache/torch/hub/checkpoints/alexnet-owt-7be5be79.pth 0%| | 0.00/233M [00:00<?, ?B/s] 86%|████████▌ | 199M/233M [00:01<00:00, 209MB/s] 100%|██████████| 233M/233M [00:01<00:00, 215MB/s] Loading model from: /usr/local/lib/python3.10/site-packages/lpips/weights/v0.1/alex.pth loding check from: /home/user/app/sam_diffsr/weight/model_ckpt_steps_400000.ckpt Traceback (most recent call last): File "/home/user/app/app.py", line 16, in <module> sam_diffsr_infer = sam_diffsr_demo() File "/home/user/app/sam_diffsr/infer.py", line 21, in __init__ self.model_init(ckpt_path) File "/home/user/app/sam_diffsr/infer.py", line 68, in model_init self.load_checkpoint(ckpt_path) File "/home/user/app/sam_diffsr/infer.py", line 60, in load_checkpoint self.model.model.cuda() File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 911, in cuda return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 802, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 802, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 825, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 911, in <lambda> return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 302, 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: