TensorFlow训练中保存tensorboard中cupti丢失问题解决

bigplapla

于 2021-03-19 15:56:27 发布

阅读量5.6k

点赞数 6

分类专栏： TensorFlow 深度学习文章标签： tensorflow 深度学习

本文链接：https://blog.youkuaiyun.com/bigplipli/article/details/115003410

版权

问题描述

在用TensorFlow进行训练的时候，往往需要使用tensorboard记录训练中的参数变化。加上tf.keras.callbacks.TensorBoard()后，发现终端输出如下：

2021-03-19 12:05:29.872424: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1456] Created TensorFlow device (/job:localhost/replica:0/task:0/device:G
PU:0 with 21676 MB memory) -> physical GPU (device: 0, name: NVIDIA GeForce RTX 3090, pci bus id: 0000:01:00.0, compute capability: 8.6)
2021-03-19

最低0.47元/天解锁文章