避免标准上的tensorflow打印

2024-06-14 15:19:46 发布

您现在位置:Python中文网/ 问答频道 /正文

任何人都知道是否有一种方法可以防止tensorflow用gpu的内存分配日志污染标准错误?。 我注意到当执行以下命令时:

with tf.Session() as sess:

tensorflow在标准错误上打印关于内存和gpu资源分配的日志。类似于:

I tensorflow/core/common_runtime/local_device.cc:25] Local device intra op parallelism threads: 48
Graphics Device pciBusID 0000:02:00.0
Free memory: 11.75GiB
...

出于重要原因,我想避免这种印刷。


Tags: 方法内存core命令标准gpusessiondevice