pytorch1.1踩坑记录
-
from tensorboardX import SummaryWriter
ImportError: No module named tensorboardX
解决:pip install tensorboardX
结果:安装成功后继续报错,
logdir: Optional[str] = None,
^
SyntaxError: invalid syntax
原因:tensorboardX和pytorch版本不对应,百度后。网上说PyTorch 1.1.0及以前的版本,最好使用 tensorboardX 1.7及以前,所以 pip install tensorboardX==1.7,成功 -
AttributeError: ‘Tensor’ object has no attribute ‘bool’
出错代码:A_mask = torch.eye(t-1, t-1).cuda().bool()
原因:pytorch1.1版本没有bool属性
解决方法:A_mask = torch.eye(t-1, t-1).cuda().byte()