今天打开电脑,打算用 GPU 跑模型时,提示 cuda 不存在,重启电脑还是连不上,网上搜索各种方法,最终解决,参考:https://www.cdsy.xyz/computer/hardware/240701/cd62027.html。
好记性不如烂笔头,这里记录一下,避免以后再遇到这种问题,希望这个方法也能帮助到同样遇到这个问题的你。
(1)终端查看显卡驱动
(2)查看CUDA,还在
(1)首先,查看自己安装的nvidia版本,我的是450.57
ls /usr/src | grep nvidia
(2)然后,终端执行一下命令
sudo apt install dkms
sudo dkms install -m nvidia -v 450.57
(3)再次输入nvidia-smi,显示: