problem in img.unsqueeze(0).to(self.device)

minivision-ai / Silent-Face-Anti-Spoofing

静默活体检测（Silent-Face-Anti-Spoofing）

Apache License 2.0

1.36k stars 453 forks source link

problem in img.unsqueeze(0).to(self.device) #95

Closed AlienX456 closed 2 years ago

AlienX456 commented 2 years ago

Hi,

I have a problem when trying to run test script, when it is trying to run this line img = img.unsqueeze(0).to(self.device) execution gets blocked. ¿Do you know what could be happening?

Thanks!!!

AlienX456 commented 2 years ago

I get cuDNN error: CUDNN_STATUS_MAPPING_ERROR after a while, it 's related with cuda, what version should i use?

Nurmukhammeds commented 2 years ago

it depends on your tensorflow version in local env that must satisfy to cuda, cudnn