Pushing tensors to the gpu hangs. For example:
import torch

a = torch.zeros(10)
a.cuda()  # hangs indefinitely 

I have tried reinstalling CUDA 8.0 and CUDNN 5.1 but this has not resolved the issue.
Some quick specs that may be handy:
os: Ubuntu 16.04
python distro: python 2.7 (conda)
gpu: nvidia gtx 1080ti
nvidia driver: 375