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