torch系列：如何在torch内使用tensorboard

Posted 2022-11-30 xizero00

torch也是可以使用tensorboard的，通过安装crayon就可以

下面以ubuntu下的安装为例进行讲解

其实安装的过程还是会碰到不少曲折的过程的

主要为

①安装crayon会提示找不到libssl.so文件，因此可以加入OPENSSL_LIBDIR=/usr/lib/x86_64-linux-gnu/来告诉luarocks，ssl库文件的路径。

②docker下载镜像的时候会使用国外的地址，因此需要更换docker的镜像，

下面给出安装过程：   （1）安装crayon的torch端
luarocks install crayon OPENSSL_LIBDIR=/usr/lib/x86_64-linux-gnu/


（2）安装docker
sudo apt install docker.io


（3）设置docker的镜像为国内的镜像https://registry.docker-cn.com
sudo vim /etc/docker/daemon.json
贴入如下：

"registry-mirrors": ["https://registry.docker-cn.com"]


（4）重启docker服务
sudo service docker restart

（5）下载镜像crayon的服务端的镜像
sudo -H docker run -d -p 8888:8888 -p 8889:8889 --name crayon alband/crayon

（6）在lua里面添加代码，从而记录loss或者accuracy

下面是样例：https://github.com/torrvision/crayon/blob/master/client/lua/README.md#usage-example

参考：

https://github.com/torrvision/crayon