Error when I try to do the inference

[JoseMoFi]

Error when I try to do the inference

[ardasatata]

@JoseMoFi I suggest you go straight install Ubuntu rather than wasting your time to set this up on W10 (been there myself & I ended up installing Ubuntu 😢)
This code works well on Ubuntu, even on the Nvidia DGX-1 environment ✌🏼

[JoseMoFi]

Ok, I am secure that the problem was WSL 2. However, I don't know if it's because I have bad config CUDA or if WSL can't work with the graphic card. But I use other code that neither work in WSL but it can work on server with Ubuntu. So I can say thay my problem is caused by WSL.
Thank you for the help!

[ycmin95]

Hi, @JoseMoFi ,
it seems like its about your environment setting, because the error occurs in the forward of ResNet. Perhaps you can check your envionment first and then an input validation may be helpful.

[JoseMoFi]

I use WSL 2, could it be the problem? And thank you for the help!

[ycmin95]

I'm not familar with WSL 2, all experiments are conducted on ubuntu. Can WSL 2 detect the GPU device?

[JoseMoFi]

Yes, WSL 2 can detect the GPU device. However, I think the problem should be WSL 2 because I had similar error in other repo when I was training and now I test again but in W10 and it work, so... I'll do more test, but it is very probable who the problem must be WSL 2 or some config.
If I find something I'll post here. And really thank you for the help!