pytorch相关可视化内容--HiddenLayer

本文介绍了如何在PyTorch环境中利用hiddenlayer模块结合graphviz进行网络可视化,包括了graphviz模块的conda安装、hiddenlayer的pip安装及实际操作中的问题解决过程,展示了从安装到测试的完整步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

pytorch相关可视化的方式有好几种,这边选择尝试最简单的HiddenLayer:

参考:pytorch中使用hiddenlayer包+安装graphviz实现可视化网络_Chuckie今天也要学习!的博客-优快云博客_hiddenlayer 安装

1.安装graphviz模块

网上很多方式是直接去下载安装包安装再进行配置环境变量实现的,我这边参考博主,直接用conda来安装:conda install python-graphviz

奇怪,让我升级,还要安装这么多包?咱也不懂,就装呗

安装结束, 奇怪,以前是torch的1.12.1版本的,现在直接变成pytorch了,没有torch了,torchvision还降低了版本了,所幸还能之前,而且之前下面的的warning也没有了。。后面发现之前博主的环境是conda创建的pytorch1.7虚拟环境中进行的,所以自动升级了把

D:\ProgramData\Anaconda3\envs\pytorchenv\lib\site-packages\torchvision\io\image.py:13: UserWarning: Failed to load image Python extension: 
  warn(f"Failed to load image Python extension: {e}")

也不晓得 python-graphviz和graphviz啥关系,反正也装了。

2.安装hiddenlayer模块

直接使用pip install hiddenlayer

这个安装的好快,很小

 3.测试

都安装完成了,测试一下,参考这篇:PyTorch学习笔记:训练过程的可视化(hiddenlayer库)_code_carrot的博客-优快云博客_hiddenlayer库

但是拷贝过来后报错,虚拟环境下执行命令安装库:conda install scikit-learn

 安装完毕

 重启下pycharm,没有报错了,再直接执行看看,又报错了 

继续安装,安装结束重启下pycharm

 太棒了,终于报缺少数据集错误了,等太久

 改成true,再执行下,出来了,但是我电脑好像卡死了。。 电脑太差没办法。。

那最后试验算是成功咯!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值