一、如何确认自己安装的 tensorflow 版本(CPU或者GPU)
完美解决查看自己使用的tensorflow是cpu还是gpu版本_源代码杀手的博客-CSDN博客_查看tensorflow-gpu版本
例如在 Pycharm中配置好 TensorFlow编译环境中新建Python文件,输入以下代码
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
运行,即可得到自己的tensorflow信息
二、安装 GPU 版本TensorFlow
1、安装显卡驱动
Ubuntu 16.04安装NVIDIA驱动_乐多的博客-CSDN博客_ubuntu2004安装nvidia驱动
(1)根据自己的显卡和系统下载完驱动之后,禁用nouveau第三方驱动
打开编辑配置文件: sudo gedit /etc/modprobe.d/blacklist.conf
在最后一行添加:blacklist nouveau
改好后执行命令:sudo update-initramfs -u
重启使之生效:reboot
(2)经过以上步骤之后,Ubuntu就无法启动了!!!!!!!
(3)在虚拟机中Ubuntu的显卡驱动是 虚拟显卡,所以不能直接安装英伟达显卡驱动,否则Ubuntu将无法进入系统,无法工作...
2、恢复
(1)在 recovery 模式下进入boot,打开 blacklist 文件,修改之前加入的条目,参考下面博客
ubuntu禁用自带nouveau驱动导致不能进入系统的解决方法_wei_supreme的博客-CSDN博客_ubuntu禁用nouveau后进不了系统
(2)如果已经按照某些教程安装了英伟达的驱动,这时一定要卸载,否则经过上面的操作也无法成功,参考下面博客方法二的 第一步和第二步再reboot即可。
Ubuntu登录报错/dev/sda1:clean的解决记录贴_lipengfei0427的博客-CSDN博客_/dev/sda2 clean