How to fix this problem?

Question

Traceback (most recent call last):
File "C:\Users\zheni\Downloads\ERDS-Pytorch-master\ERDS-Pytorch-master\train_CNN_pooled_improved.py", line 124, in <module>
model = CNN2D(input_size = input_size,
File "C:\Users\zheni\Downloads\ERDS-Pytorch-master\ERDS-Pytorch-master\nu_models.py", line 42, in __init__
self.batchnorm.append(nn.BatchNorm2d(jj, eps=0.001, momentum=0.01,
File "C:\Users\zheni\AppData\Local\Programs\Python\Python38\lib\site-packages\torch\nn\modules\module.py", line 491, in cuda
return self._apply(lambda t: t.cuda(device))
File "C:\Users\zheni\AppData\Local\Programs\Python\Python38\lib\site-packages\torch\nn\modules\module.py", line 409, in _apply
param_applied = fn(param)
File "C:\Users\zheni\AppData\Local\Programs\Python\Python38\lib\site-packages\torch\nn\modules\module.py", line 491, in <lambda>
return self._apply(lambda t: t.cuda(device))
File "C:\Users\zheni\AppData\Local\Programs\Python\Python38\lib\site-packages\torch\cuda\__init__.py", line 170, 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

Answer 1

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

Answer 2

not CPU (Central Point Unit), but GPU (Graphic Point Unit)...

have you an NVIDIA graphic card / chipset on your device?

if yes, have you installed the driver available at the link provided?

if no, bad luck: it seems that the needed "torch" package requires one to work ^^

Answer 3

visph but I put cpu option strange