python: 3.9.13

torch:1.13.1

torchvision:0.14.1

pip install hiddenlayer

 安装失败。

再次使用下列方式:

pip install hiddenlayer==0.3 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装成功!

但需要导入graphviz模块才能进行图形的显示,hiddenlayer模块中调用了graphviz模块

但使用时报错:

module 'torch.onnx' has no attribute '_optimize_trace'

可能的原因:由于pytorch版本较新,hiddenLayer内部的API没有相应地更新,HiddenLayer还是调用的_optimize_trace,而新版pytorch已经改成了_optimize_graph。

按照网上的方法将pytorch_builder.py第71行中的_optimize_trace更改为_optimize_graph。解决了此问题。但又出现下一个问题:

ModuleNotFoundError: No module named 'graphviz'

然后利用安装graphviz模块,在运行又报错

安装conda install graphviz,不成功;安装pip install graphviz,不成功

使用pip install -i https://pypi.tuna.tsinghua.edu.cn/simple graphviz

 可以了。

备注:以下为尝试过程中一些没成功的片段,也保留下来,方便日后查阅。

 pip install torchviz

 没成功。

下载

graphviz

Download | Graphviz

要设置系统环境变量,放在了C盘anaconda3下的Graphviz的bin中 

C:\Users\QHN\anaconda3\lib\site-packages\hiddenlayer\pytorch_builder.py:83: FutureWarning: 'torch.onnx._patch_torch._node_getitem' is deprecated in version 1.13 and will be removed in version 1.14. Please Internally use '_node_get' in symbolic_helper instead..
  params = {k: torch_node[k] for k in torch_node.attributeNames()}

Logo

永洪科技,致力于打造全球领先的数据技术厂商,具备从数据应用方案咨询、BI、AIGC智能分析、数字孪生、数据资产、数据治理、数据实施的端到端大数据价值服务能力。

更多推荐