On a fresh install of Ubuntu 16.04 LTS, using the recommended install:
conda install pytorch torchvision cuda80 -c soumith

PyTorch works immediately. But a call to my_tensor.cuda() results in:
AssertionError:
Found no NVIDIA driver on your system. Please check that you
have an NVIDIA GPU and installed a driver from
http://www.nvidia.com/Download/index.aspx

Installing CUDA separately works. But according to this thread it seems that PyTorch’s built in CUDA should work without a separate installation. I’ll be allocating and deallocating many machines running PyTorch, so I just wanted to check if the manual CUDA installation is expected at the moment? Or is there something I’m missing? Thank you!