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…).