tensorboardX使用中 AttributeError: 'function' object has no attribute 'graph'

最近在使用tensorboardX可視化網絡結構，因為tensorboardX並非pytorch原生的可視化工具，所以版本之間並不兼容

在使用的過程中會遇到，AttributeError: 'function' object has no attribute 'graph'，大概是版本之間不兼容

完整錯誤如下

Traceback (most recent call last):
  File "source_hourglassnet.py", line 181, in <module>
    writer.add_graph(model, (input_data, ))
  File "/home/yongjie/anaconda2/envs/danfeng/lib/python3.6/site-packages/tensorboardX/writer.py", line 738, in add_graph
    self._get_file_writer().add_graph(graph(model, input_to_model, verbose, **kwargs))
  File "/home/yongjie/anaconda2/envs/danfeng/lib/python3.6/site-packages/tensorboardX/pytorch_graph.py", line 241, in graph
    graph = trace.graph
AttributeError: 'function' object has no attribute 'graph'

搜到了一個解決方案說的是tensorboardX1.8對於低版本的pytorch支持不是很好

conda list發現 tensorboardX確實是1.8版本的，pip uninstall tensorboardX之后再

重新安裝pip install tensorboardX=1.4

安裝完之后就可以正常使用了

沒毛病

 

但是這里有另外一個問題就是，我用pip search tensorboardX

的時候並沒有發現1.4版本的，但是我用pip install tensorboardX=1.4的時候是可以使用的，不知道為啥