首先我们需要安装tensorboard包
打开你想要的虚拟环境
conda activate 你的虚拟环境的名称
然后安装tensorboard包
pip install tensorboardX
打开我们的jupyter,输入下面代码并运行
from torch.utils.tensorboard import SummaryWriter
在当前路径下创建一个test文件夹,来存放我们的记录
writer=SummaryWriter(log_dir='test')
写入
writer.log_dir
这个仅仅是作为测试使用
for i in range(10):
writer.add_scalar('mul',i**i,i)
由于我们的tensorboard需要x86的环境,但是m1是arm架构的,所以我们需要在终端输入
arch -x86_64 zsh
进行架构的转换
然后找到我们刚刚的那个test的路径,也就是找到包含这个test的文件夹的路径,然后启动我们的服务
tensorboard --logdir="test"
然后就可以运行了
打开这个网址,就可以看到我们的可视化界面了