PyTorch使用tensorboard(遠程服務器和本地計算機的聯動)

本文基於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):

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章