caffe在训练的时候遇到 no CUDA-capable device is detected的问题
解决办法:
查看驱动版本号
cd /usr/src
如下:
这里我的版本号是418.74
sudo apt-get install dkms
sudo dkms install -m nvidia -v 418.74
然后再次输入
nvidia-smi
熟悉的界面就回来了
caffe在训练的时候遇到 no CUDA-capable device is detected的问题
解决办法:
查看驱动版本号
cd /usr/src
如下:
这里我的版本号是418.74
sudo apt-get install dkms
sudo dkms install -m nvidia -v 418.74
然后再次输入
nvidia-smi
熟悉的界面就回来了