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!