cuda 版本不一致，The NVIDIA driver on your system is too old

遇到问题

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)

解决方法

除了pytorch的torch和tochvision版本不一致以外，cuda版本也可能不一致，就算上述的版本一致，nvidia驱动的版本过旧或者过新也会导致运行报错

• 查看GPU型号

lspci | grep -i nvidia

• 查看NVIDIA驱动版本
  对应版本如440,364等可以去官网查看对应的cuda版本10还是9

sudo dpkg --list | grep nvidia-*
或者
cat /proc/driver/nvidia/version