runtime error

(6): MaxPool2d(kernel_size=3, stride=2, padding=1, dilation=1, ceil_mode=False) ) ) (conv3): SeparableConv2d( (conv1): Conv2d(1024, 1024, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=1024, bias=False) (pointwise): Conv2d(1024, 1536, kernel_size=(1, 1), stride=(1, 1), bias=False) ) (bn3): BatchNorm2d(1536, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (act3): ReLU(inplace=True) (conv4): SeparableConv2d( (conv1): Conv2d(1536, 1536, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=1536, bias=False) (pointwise): Conv2d(1536, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False) ) (bn4): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (act4): ReLU(inplace=True) (global_pool): SelectAdaptivePool2d(pool_type=avg, flatten=Flatten(start_dim=1, end_dim=-1)) (fc): Linear(in_features=2048, out_features=1000, bias=True) ) Traceback (most recent call last): File "/home/user/app/app.py", line 236, in <module> model1 = Model1(2).cuda() File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 911, in cuda return self._apply(lambda t: t.cuda(device)) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 802, in _apply module._apply(fn) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 802, in _apply module._apply(fn) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 825, in _apply param_applied = fn(param) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 911, in <lambda> return self._apply(lambda t: t.cuda(device)) File "/home/user/.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: