Al ejecutar el modelo con GPU, indica que cuda no existe, ¡se ha instalado! ¡Ningún controlador de nvidia volvió a fallar! nvidia-smi
Mire apresuradamente , sugiero los siguientes errores:
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver.
Make sure that the latest NVIDIA driver is installed and running.
Busqué en Internet y utilicé el siguiente método para resolver con éxito el problema, referencia: https://blog.csdn.net/hangzuxi8764/article/details/86572093
sudo apt install dkms
sudo dkms install -m nvidia -v 418.87.00
Entre ellos, 418.87.00 es el número de versión del controlador nvidia instalado previamente, que se puede encontrar mediante los siguientes métodos:
ls /usr/src | grep nvidia