轉載: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