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
Torch was installed using pip3 in home directory.