如何判断tensorflow是否从python shell内部使用gpu加速?
不,我认为“开放式CUDA库”不足以说明问题,因为图形的不同节点可能位于不同的设备上。
要找出使用哪个设备,您可以启用日志设备放置,如下所示:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
检查您的控制台是否有这种类型的输出。
你尚未登录,登录后可以
和开发者交流问题的细节
关注并接收问题和回答的更新提醒
参与内容的编辑和改进,让解决方法与时俱进
联系我
602392714
清零编程群