apt install nvidia-cuda-toolkit
# apt install nvidia-cuda-toolkit
but after that, the nvidia-smi cannot be working.
(launcher) root@ip-172-31-33-152:~# nvidia-smi
Failed to initialize NVML: Driver/library version mismatch
So, I delete nvidia-cuda-toolkit again.
(launcher) root@ip-172-31-33-152:~# sudo apt-get purge --auto-remove nvidia-cuda-toolkit
nvidia-smi can work now.
nvidia-cuda-toolkit install and uninstall nvidia-smi
原文：https://www.cnblogs.com/morganh/p/14674319.html