转载:https://blog.csdn.net/kudou1994/article/details/86735451
服务器在训练模型,另一边我在瞎胡乱搞不晓得咋个搞的,就不能使用GPU了。
python2.7:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
打算卸载原有的tensorflow重新安装来解决。
踩了很多坑,也尝试过强制安装TF的方法,最后还是不行。
最后怀疑是没有卸载干净,又完完全全的卸载了一遍,最后得以解决
conda uninstall tensorflow
conda uninstall tensorflow-gpu
最后重新安装tensorflow-gpu即可 #不需要安装CPU版本的tensorflow
conda install tensorflow-gpu