UserWarning: CUDA initialization: Found no NVIDIA driver on your system

OS - Linux, Python version - 3.6.8, Torch version - 1.7.1,
NVIDIA driver - 440.33, CUDA - 10.2, GPU card version - NVIDIA Volta T100.

When I try to use PyTorch in interactive Python session this error message pops up:
β€œ/home/gsergei/.local/lib/python3.6/site-packages/torch/cuda/init.py:52: UserWarning: CUDA initialization:
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 (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:100.)
return torch._C._cuda_getDeviceCount() > 0
False”

Torch was installed using pip3 in home directory.

This error points towards an issue with your driver, which might be caused by e.g. a driver update without a restart etc.

1 Like