欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

tensorflow不能调用GPU

时间:2023-05-20

问题:在环境了安装了tensorflow-gpu,但是,在程序运行时,代码并没有调用GPU

原因:环境中存在CPU版tensorflow,程序运行时可能默认使用CPU版tensorflow。

解决方法:卸载环境中所有的tensorflow包,然后重新安装tensorflow-gpu

步骤1:查看环境中安装的所有的包

pip list

步骤2:卸载cpu版tensorflow

pip uninstall tensorflowpip uninstall tensorflow-gpu

步骤3:仅安装GPU版tensorflow

pip install tensorflow-gpu==1.14或conda install tensorflow-gpu==1.14

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。