I have the same issue.
Possible solution is to download the version of pytorch from the Prelims section of document ( !pip install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl ...), meaning pytorch version 0.3.0 on cuda version 8.
I haven’t confirmed this works, though. Downgrading proved a bit too much of a headache for me to think it’s worth it. Trying to find alternative solutions.