from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('torch_tensorboard_example')
使用上面两行代码的时候,报错找不到tensorboard。
ModuleNotFoundError: No module named 'tensorboard'
解决方法
使用pip install tensorflow
在pytorch中安装tensorflow(同时默认安装我们需要的tensorboard)
既然我们是直接使用pytorch的
torch.utils.tensorboard
,那么我们为什么还要安装tensorflow呢?因为pytorch内部的代码实现就是用的tensorflow,所以我们必须安装tensorflow