ModuleNotFoundError: No module named ‘caffe2‘

在深度学习训练中,调用tensorboard时,有时会出现以下错误:

ModuleNotFoundError: No module named 'caffe2'

但是,该项目代码中并没有使用到caffe2.此问题可通过将引用包中的:

# 更改前
from torch.utils.tensorboard import SummaryWriter

更改为以下引用方式,即可解决问题.

# 更改后
from tensorboardX import SummaryWriter