刚开始我是用 gpu 去训练的模型,后来发现训练出来的模型识别不出来物体最后仔细观察了一下发现红框里面的参数是 nan。
然后用 cpu 去训练了一下红框里就有参数了,
有人知道为什么我用
gpu去跑这些参数是nan 吗?
sudo apt-get remove --purge nvidia-* (这个是使用自动安装的卸载方式)
sudo ./NVIDIA-Linux-x86_64---xxx.run --uninstall (这个是使用.run 文件安装的卸载方式,xxx 代表版本号)
!开始安装
sudo service lightdm stop (关闭图形界面)
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-driver-460
reboot
进入图形界面
nvidia-smi
查看是否安装成功
这个是这个