如何在导入 TensorFlow 时禁用额外的 Cuda 信息

如何在导入 TensorFlow 时禁用额外的 Cuda 信息

当我在 tensorflow 环境中输入与 cuda 相关的 python 命令时,我得到了一些有关 cuda 的信息,我该如何禁用/启用它们?

>>> import tensorflow as tf
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so locally

相关内容