安装HiddenLayer可视化网络、PyTorchViz可视化网络
安装hiddenLayer、graphviz模块
·
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
要设置系统环境变量,放在了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()}
更多推荐
所有评论(0)