Hi,
I would expect that you have an old cuda8.0 pytorch install lingering somewhere that is being picked up instead of what you install.
Make sure to uninstall all pytorch / pytorch-nightly packages before installing the new one.