0
点赞
收藏
分享

微信扫一扫

tensorflow不能调用GPU

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

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

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

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

pip list

步骤2:卸载cpu版tensorflow

pip uninstall tensorflow
pip uninstall tensorflow-gpu

步骤3:仅安装GPU版tensorflow

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

相关推荐

0 条评论