This also happens on my system

Ubuntu 16.04
Nvidia GeForce 940MX
PyTorch 0.3.1 running on Python 3.6
Cuda 8.0
CUDNN 7

Any clues? I don’t see why this thread is marked as resolved, if the solution is to restart the laptop every time.