之前程序可以运行,此次打开报错:
RuntimeError: No CUDA GPUs are available
检查驱动和cuda是否可用
nvidia-smi:输出版本号,为可用
nvcc -V:输出版本号,为可用
上述显示驱动不可用,先不着急重装,下面链接可以尝试更新驱动
DKMS可以帮我们维护内核外的这些驱动程序,在内核版本变动之后可以自动重新生成新的模块。
sudo apt-get install dkms
sudo dkms install -m nvidia -v 450.80.02
参考以下方法: