服务器在训练模型,另一边我在瞎胡乱搞不晓得咋个搞的,就不能使用GPU了。
python3.6下
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
查询了一下确实检测不到GPU了,只有CPU。
随后使用
pip list
pip3 list
都查询了一下版本也都正常
tensorflow和tensorflow-GPU都是1.12.0版本的(如果想使用GPU则tensorflow-gpu的版本一定不能比tensorflow的版本低)
另外也没有安装tf-lights之类的(之前被坑过)
最后打算卸载原有的tensorflow重新安装来解决。
踩了很多坑,也尝试过强制安装TF的方法,最后还是不行。
最后怀疑是没有卸载干净,又完完全全的卸载了一遍,最后得以解决
sudo pip uninstall protobuf #对应自己的安装方式进行卸载
sudo pip3 uninstall protobuf
pip3 uninstall protobuf
pip uninstall protobuf
sudo pip uninstall tensorflow
sudo pip3 uninstall tensorflow
pip uninstall tensorflow
pip3 uninstall tensorflow
sudo pip uninstall tensorflow-gpu
sudo pip3 uninstall tensorflow-gpu
pip uninstall tensorflow-gpu
pip3 uninstall tensorflow-gpu
最后重新安装tensorflow-gpu即可 #不需要安装CPU版本的t