目录

构建可用的JupyterLab和TensorBoard

  • 启动YOLOv5容器
    docker run --ipc=host --runtime nvidia -it -p 8888:8888 \
      -v ${dataset_dir}:/usr/src/app/project \
      ultralytics/yolov5:latest
    
  • 安装版本1的TensorBoard。(解决FAQ1的问题:jupyter-tensorboard 0.2.0不支持高于TensorBoard 2.0的版本。YOLOv5镜像中安装的TensorBoard 2.4的版本。)
    pip uninstall tensorboard -y && pip install tensorboard==1.15
    
  • 运行JupyterLab
    jupyter lab --no-browser --ip 0.0.0.0 --port 8888
    
  • 本地浏览器进行访问
    http://ip:8888/lab
    

FAQ

  1. Launcher Error - Invalid response: 500 Internal Server Error
    Uncaught exception POST /api/tensorboard?1609481325314 (192.168.1.1)
     HTTPServerRequest(protocol='http', host='hostname:20101', method='POST', uri='/api/tensorboard?1609481325314', version='HTTP/1.1', remote_ip='192.168.1.1')
     Traceback (most recent call last):
       File "/opt/conda/lib/python3.8/site-packages/tornado/web.py", line 1702, in _execute
         result = method(*self.path_args, **self.path_kwargs)
       File "/opt/conda/lib/python3.8/site-packages/tornado/web.py", line 3173, in wrapper
         return method(self, *args, **kwargs)
       File "/opt/conda/lib/python3.8/site-packages/jupyter_tensorboard/api_handlers.py", line 39, in post
         self.settings["tensorboard_manager"]
       File "/opt/conda/lib/python3.8/site-packages/jupyter_tensorboard/tensorboard_manager.py", line 219, in new_instance
         create_tb_app(
       File "/opt/conda/lib/python3.8/site-packages/jupyter_tensorboard/tensorboard_manager.py", line 44, in create_tb_app
         return application.standard_tensorboard_wsgi(
     AttributeError: module 'tensorboard.backend.application' has no attribute 'standard_tensorboard_wsgi'
    

参考资料