I am not sure I can help you all the way, but did you check your graphics card, if it has CUDA performances and whether you had the most recent drivers for it? I think pytorch requires
NVIDIA CUDA 7.5 or above
NVIDIA cuDNN v6.x or above
Otherwise, maybe this might help you :
(I should warn that I am not using Windows myself so this is the extent of the help I might offer ^^’)
Yes, I can use CUDA with tensorflow, but I hate it and I wanted to use pytorch instead. All the required dependencies have been installed, and in fact I can run pytorch quite well. The problem is that my graphics card is not supported in the published binaries and I need to build from source to get support for my old GPU, but the only instructions on how to build from source are for linux and OSX and they fail in windows because of syntax.
My GPU has CUDA compute capabilities of 5, and I wasn’t able to get it to work correctly with the Pytorch 0.4.x binaries on Anaconda, but I did manage to compile it in Windows. Here’s the raw footage of how I did it:
Note that in the video, I had to restart several times due to errors and I purposely left them in there in case you’ve also encountered them. There are no guarantees that these are the only errors you’ll encounter.