tensorflow不断升级,最新版本的代码需要更高版本的cuda支持。以前一直用cuda8.0和cudnn6.0的组合,tensorflow用1.3.0版本。不过最新下载的models,某些代码执行时,已经出现了错误。
类似的错误是因为tensorflow升级,models中调用了新版本的函数,而安装的tensorflow是旧版本。为了执行代码,需要更新tensorflow至新版本。而tensorflow新版本,又要求更高版本的cuda和cudnn。因此这里简单说一下卸载nvidia驱动和cuda的过程,至于重新安装,可以参考博客https://blog.csdn.net/ezhchai/article/details/69663082
1.卸载nvidia驱动
卸载NV驱动和安装一样,首先ctrl+Alt+F2进入命令行状态,然后停止lightdm
sudo service lightdm stop
或者
sudo /etc/init.d/lightdm stop
卸载命令位置/usr/bin/nvidia-uninstall,以下命令即可卸载。
sudo /usr/bin/nvidia-uninstall
不找这个命令的位置,也可以
sudo apt-get install autoremove --purge nvidia*
这样就完成了卸载。
2.卸载cuda
cuda的默认安装在 /usr/local/cuda-8.0下,用下面的命令卸载:
sudo /usr/local/cuda-8.0/bin/uninstall_cuda-8.0.pl
会出现长时间的刷屏现象,说明cuda文件其实还蛮多的~~
/usr/local/cuda-8.0没有自动删除,需要手动完成删除。其实删不删无所谓,知道它没用了也就是了。
卸载之后,重启一下。可以看到分辨率有所不同,这是按照https://blog.csdn.net/ezhchai/article/details/69663082中描述的方法,重新安装cuda9.0和cudnn7.0即可。