#from torch.utils.tensorboard import SummaryWriter
tensorflow已经安装了,顺带说一句安装tensorflow后好像自动会有自适应的匹配版本的tensorflow-tensorboard,好像不用管tensorboard了。。
但是他还是报错,所以改了一下:
from tensorboardX import SummaryWriter
就行了,没安装tensorboardx的安装一下,注意大小写:
pip install tensorboardx