问题描述
最近看到pytorch更新后的1.1.0版本原生支持tensorboard,但是在安装1.1.0版本之后,导入tensorboard时出现了错误,错误信息如下:
错误原因
目前tensorboard释放的稳定版只支持到1.13,通过报错的最后一行可以看出,需要1.14及以上的版本
问题解决
所以,要解决这个问题,需要安装tenforboard的nightly版本,需要安装tb-nightly,因为我是conda环境下安装的pytorch,所以也想用conda安装,但是发现conda找不到tb-nightly这个包,只能通过pip来安装,运行一下命令,
- pip install tb-nightly
安装之后再次import torch.utils.tensorboard,发现还是报错,错误信息如下
因为这里还要安装一个future包(我也不知道这个包是干嘛的,哈哈 ),好的,继续安装
- pip install future
安装完了验证一下,成功导入!