Solved by tomahawk810 in post #2 
              
              
                The above issue was happening using Pytorch installed using pip. 
I switched to a Conda installation, using the Python 3.7/Cuda 9.0 version. 
And this solved the problem for me (I don’t know why exactly…).