您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何判断tensorflow是否从python shell内部使用gpu加速?

如何判断tensorflow是否从python shell内部使用gpu加速?

不,我认为“开放式CUDA库”不足以说明问题,因为图形的不同节点可能位于不同的设备上。

要找出使用哪个设备,您可以启用日志设备放置,如下所示:

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

检查您的控制台是否有这种类型的输出

python 2022/1/1 18:37:23 有225人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶