以下是一些CentOS上PyTorch的可视化工具推荐:
pip install hiddenlayer,可通过build_graph方法生成网络结构图并保存为图片。pip install torchviz,使用make_dot函数生成可视化图。pip install tensorboard,需在代码中使用SummaryWriter记录数据,启动后可在浏览器查看。pip install tensorboardX,用法与TensorBoard类似。pip install netron,通过netron.show方法展示模型。