caffe在训练的时候遇到 no CUDA-capable device is detected的问题
在这里插入图片描述

解决办法:
查看驱动版本号

cd /usr/src

如下:
查看Nvidia驱动版本号
这里我的版本号是418.74

sudo apt-get install dkms
sudo dkms install -m nvidia -v 418.74

然后再次输入

nvidia-smi

熟悉的界面就回来了
在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐