When I try to run a resnet, I met this problem, and don’t know how to solve it.
My gpu is GeForce GTX 780M, cuda9.1, win10, vs2017.
Should I change to cuda8.0?
Thanks a lot!
Did you try run anything except resnet? Something very basic, for example.
You can try with CUDA 8.0 and I believe it will solve your problem. Otherwise if you really need support of CUDA 9.0 or CUDA 9.1, drop me a line and I will help you to compile PyTorch from source with support for your GPU. Just did it yesterday for my GTX 760.
When I changed to CUDA 8.0, my problem was solved.
Thank you!