在YOLOv5中运行JupyterLab和TensorBoard
构建可用的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 Launcher Error - Invalid response: 500 Internal Server Error Uncaught exception POST /api/tensorboard?1609481325314 (192.168.1.