Encounter problems
The NVIDIA driver on your system is too old (found version 9010).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.
(mmlab) root@bda240ba6e9c:/home/41228/zlk/VQA_DL# cat /proc/driver/nvidia/version
NVRM version: NVIDIA UNIX x86_64 Kernel Module 387.26 Thu Nov 2 21:20:16 PDT 2017
GCC version: gcc version 4.8.5 20150623 (Red Hat 4.8.5-16) (GCC)
Solution
In addition to the inconsistency of pytorch's torch and tochvision versions, the cuda version may also be inconsistent. Even if the above versions are the same, the version of the nvidia driver is too old or too new to cause an error when running
- View GPU model
lspci | grep -i nvidia
- Check the NVIDIA driver version. The
corresponding version such as 440, 364, etc. can go to the official website to check the corresponding cuda version 10 or 9
sudo dpkg --list | grep nvidia-*
或者
cat /proc/driver/nvidia/version