Yes, I run into error with your code snippet. Do you know how to build cuda from source in windows10? I see the guide on pytorch and it seems that build from source in ubuntu is easy, but build in windows 10 is complicated.