遇到问题
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