An error occurred when I run the test.py

[yygbruce]

os:ubuntu18.04 cuda:10.02 the error: Traceback (most recent call last): File "test.py", line 88, in evaluate(dspth='/home/face-parsing/data/', cp='79999_iter.pth') File "test.py", line 74, in evaluate out = net(img)[0] File "/home/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, kwargs) File "/home/face-parsing/model.py", line 243, in forward feat_res8, feat_cp8, feat_cp16 = self.cp(x) # here return res3b1 feature File "/home/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, *kwargs) File "/home/face-parsing/model.py", line 106, in forward feat8, feat16, feat32 = self.resnet(x) File "/home/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(input, kwargs) File "/home/face-parsing/resnet.py", line 73, in forward x = F.relu(self.bn1(x)) File "/home/.local/lib/python2.7/site-packages/torch/nn/functional.py", line 914, in relu result = torch.relu(input) RuntimeError: CUDA error: no kernel image is available for execution on the device