本文基於Ubuntu16.04系統,Python3.7
---------------------------------------------------------------------------------
首先是配置PyTorch和tensorboard的版本
1. PyTorch >= 1.1.0 版本,
2. tensorboard版本 >= 1.1.15 (PyTorch使用的tensorboard於tensorflow使用的無異)
查看上述兩個東西的版本的方法:
PyTorch
tensorboard
如果tensorboard的版本不夠高,請按如下pip安裝nightly build版本的tensorboard(1.15.0版本目前):
$ pip install tb-nightly
----------------------------------------------------------------------------------------------------------------------
----------------------------------------------------------------------------------------------------------------------
-
版本配置好後,啓動tensorboard
在你正在使用的project的文件夾中,啓動tensorboard:
$ tensorboard --logdir=runs
確保在該project文件夾中有一個名叫runs的子文件夾,用來存放Summarywriter寫入的文件。
啓動tensorboard之後,tensorboard的UI界面可以在瀏覽器中使用如下url打開:
http://localhost:6006
如果是在遠程服務器上啓動tensorboard,但是希望在本地電腦的瀏覽器中查看該tensorboard的內容,則請做如三步:
1. 建立遠程服務器於本地電腦的連接:
$ ssh -N -f -L localhost:8888:localhost:8889 username@your_remote_host_name比如
我的配置就是:
$ ssh -N -f -L localhost:8888:localhost:8889 [email protected]
2. 在遠程服務器上啓動tensorboard
$ tensorboard --logdir=runs
3. 在本地打開如下url(與http://localhost:6006不同了)
http://172.1x.1x.xxx:6006 前面是遠程服務器的IP
----------------------------------------------------------------------------------------------------------------------
程序中用Summarywriter把需要展示的內容寫入文件,存入./runs 文件夾
1. import:(此處僅展示與tensorboard相關的import)
2. setup
3. In the training loop:
----------------------------------------------------------------------------------------------------------------------
本地瀏覽器中查看效果如下(FashionMNIST dataset):