AssertionError: Torch not compiled with CUDA enabled

[ashwath1295]

AssertionError Traceback (most recent call last) ~\AppData\Local\Temp/ipykernel_18688/1973085970.py in 53 54 ---> 55 net = EEGNet().cuda(0) 56 print (net.forward(Variable(torch.Tensor(np.random.rand(1, 1, 120, 64)).cuda(0)))) 57 criterion = nn.BCELoss()

~\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py in cuda(self, device) 678 Module: self 679 """ --> 680 return self._apply(lambda t: t.cuda(device)) 681 682 def xpu(self: T, device: Optional[Union[int, device]] = None) -> T:

~\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py in _apply(self, fn) 568 def _apply(self, fn): 569 for module in self.children(): --> 570 module._apply(fn) 571 572 def compute_should_use_set_data(tensor, tensor_applied):

~\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py in _apply(self, fn) 591 # with torch.no_grad(): 592 with torch.no_grad(): --> 593 param_applied = fn(param) 594 should_use_set_data = compute_should_use_set_data(param, param_applied) 595 if should_use_set_data:

~\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\nn\modules\module.py in (t) 678 Module: self 679 """ --> 680 return self._apply(lambda t: t.cuda(device)) 681 682 def xpu(self: T, device: Optional[Union[int, device]] = None) -> T:

~\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\cuda_init_.py in _lazy_init() 206 "multiprocessing, you must use the 'spawn' start method") 207 if not hasattr(torch._C, '_cuda_getDeviceCount'): --> 208 raise AssertionError("Torch not compiled with CUDA enabled") 209 if _cudart is None: 210 raise AssertionError(

AssertionError: Torch not compiled with CUDA enabled