geyuying/PF-AFN

Test using CPU

bondbenz opened this issue · 4 comments

bondbenz commented

Hello , i want to test the pretrained model using only CPU but asking always for cuda even that I disabled cuda device.

cloudy4next commented

@bondbenz, If i'm not wrong, you can't test on cpu cause the matrix is used to calucate it's not numpy , so it's require cupy and cupy needs gpu to performe calculation 👍🏾

bondbenz commented

Oh shoot thats bad , i tried to replace cuda and cupy and load model using cpu but still nothing keep getting cuda device unavailable.

  • 👍1
cloudy4next commented

@bondbenz cupy is intregated with cuda itself for some corelation loss or something . try with a gpu ✌🏾
you may close the issue now

  • 👍1
bondbenz commented

Thank you for your replies