Tensorflow不能使用GPU的解決辦法


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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM