torchviz进行pytorch神经网络可视化

一、安装

需要系统安装Graphviz工具，如果是自己电脑上可以用yum、apt等工具安装，但是如果在服务器上，没有root权限的那种就会比较困难。需要root权限，如果是虚拟的容器可以在dockerfile中进行安装。

在pip安装的时候torchviz可能会重新安装torch，可能会造成torch和cuda等环境不匹配的问题。如果不想重新安装torch，可以pip后加个参数--no-deps，就不会安装依赖库了

二、使用

y = model(batch_x)
MyConvNetVis = make_dot(y, params=dict(list(model.named_parameters()) + [('x', batch_x)]))
MyConvNetVis.format = "png"
# 指定文件生成的文件夹
MyConvNetVis.directory = "data"
# 生成文件
MyConvNetVis.view()