如何解决如何知道Keras使用的是GPU还是CPU
有什么方法可以检查Keras框架是否使用GPU或CPU训练模型?
我正在使用keras在GPU上训练我的模型,但是它是如此之慢,以至于我不确定是在使用CPU还是GPU进行训练。
解决方法
首先,您需要找到GPU设备:
physical_device = tf.config.experimental.list_physical_devices('GPU')
print(f'Device found : {physical_device}')
然后您可以使用以下代码检查GPU设备是否处于“用于训练”状态:
tf.config.experimental.get_memory_growth(physical_device[0])
如果此代码返回False
或什么都没有,那么您可以在下面运行此代码来设置GPU进行训练
tf.config.experimental.set_memory_growth(physical_device[0],True)
,
首先,请确保tensorflow正在检测您的GPU。运行下面的代码。如果GPU数量= 0,则表示未检测到您的GPU。为了使Tensorflow使用GPU,您需要安装Cuda工具包和Cudnn。如果未检测到GPU,并且您正在使用Anaconda,请在Conda中重新安装tensorflow。它会自动安装工具箱和Cudnn。当您使用Pip安装tensorflow时,它们不会安装它们。
import tensorflow as tf
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print(tf.__version__)
print("Num GPUs Available: ",len(tf.config.experimental.list_physical_devices('GPU')))
tf.test.is_gpu_available()
!python --version
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。