最近用tensorboardX可视化论文中的一些数据,然后找相关内容,在此记录一下。
1、第一步安装pip install tensorboardX
2、创建一个writer = SummaryWriter(log_dir='scalar')
参数log_dir=‘scalar’,表示生成的文件所存储 的路径。
3、记录数据。writer.add_scalar('loss', loss, epoch)
,第一个参数表示生成图表的名称,第二个和第三个简单理解就是图标的纵坐标和横坐标。
记录完数据以后运行发现一个Could not load dynamic library ‘cudart64_101.dll‘; dlerror: cudart64_101.dll not found
的错误,反正就是不能正确激活tensorboard。
解决方案:
1、去官网下载cudart64_101.dll。解压以后有两个文件,
2、放置cudart64_101.dll文件
然后再次运行。成功解决。