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
@H_404_6@sudo service lightdm stop
或者
@H_404_6@sudo /etc/init.d/lightdm stop
卸载命令位置/usr/bin/nvidia-uninstall,以下命令即可卸载。
@H_404_6@sudo /usr/bin/nvidia-uninstall
不找这个命令的位置,也可以
@H_404_6@sudo apt-get install autoremove --purge nvidia*
这样就完成了卸载。
2.卸载cuda
cuda的默认安装在 /usr/local/cuda-8.0下,用下面的命令卸载:
@H_404_6@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即可。
总结
如果觉得编程之家网站内容还不错,欢迎将编程之家网站推荐给程序员好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
小编个人微信号 jb51ccc
喜欢与人分享编程技术与工作经验,欢迎加入编程之家官方交流群!