今天用keras训练时发现内存占用率出奇的高,而且显存占用率出奇的低,原来keras没有用gpu训练。
通过
pip list
看到同时安装了tensorflow和tensorflow-gpu,keras默认调用了tensorflow
解决办法
同时卸载tensorflow,tensorflow-gpu,keras
再安装tensorflow-gpu和keras就可以解决
本文介绍了一种常见的Keras训练时未正确利用GPU资源的问题及其解决方案。作者遇到Keras训练过程中内存占用率异常高而显存占用率却很低的情况,发现原因是Keras默认调用了CPU版本的TensorFlow而非GPU版本。文中详细描述了解决该问题的方法:首先卸载已安装的tensorflow、tensorflow-gpu和keras,然后重新安装tensorflow-gpu和keras。
今天用keras训练时发现内存占用率出奇的高,而且显存占用率出奇的低,原来keras没有用gpu训练。
通过
pip list
看到同时安装了tensorflow和tensorflow-gpu,keras默认调用了tensorflow
解决办法
同时卸载tensorflow,tensorflow-gpu,keras
再安装tensorflow-gpu和keras就可以解决
您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型
4569

被折叠的 条评论
为什么被折叠?