OK, I didn’t know pytorch shipped with those binaries. how does this look?
will update blog if looks good:

  
      gist.github.com
  
  
    https://gist.github.com/mjdietzx/fda9535e3246f0db39b0da80403265d1 48
install-tesla-driver-ubuntu.sh
# http://www.nvidia.com/download/driverResults.aspx/117079/en-us
wget http://us.download.nvidia.com/tesla/375.51/nvidia-driver-local-repo-ubuntu1604_375.51-1_amd64.deb
sudo dpkg -i nvidia-driver-local-repo-ubuntu1604_375.51-1_amd64.deb
sudo apt-get update
sudo apt-get -y install cuda-drivers

echo "Reboot required."