PyTorch was the only framework (and I tried with Theano and TensorFlow as well) that I was able to use right with GPU. Here’s tutorial that worked for me (Win10): https://github.com/peterjc123/pytorch-scripts 1.9k