代码之家  ›  专栏  ›  技术社区  ›  grabbag

如何将TensorBoard添加到TensorFlow/TensorFlow Dockerfile

  •  0
  • grabbag  · 技术社区  · 6 年前

    我有一个简单的dockerfile,它添加了keras、枕头,但不太清楚如何添加tensorboard。我的文件:

    FROM tensorflow/tensorflow:latest-gpu-py3-jupyter
    RUN pip3 install keras
    RUN pip3 install Pillow
    RUN mkdir /tmp/logs
    CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/tf --ip 0.0.0.0 --no-browser --allow-root"]
    

    建造

    sudo docker build -t jjg .
    

    运行(8888表示Jupyter,6006表示Tensorboard)

    sudo docker run --runtime=nvidia -it --rm -v ~/Documents:/tf/notebooks -p 8888:8888 -p 6006:6006 jjg:latest
    

    手动运行张紧器板的工作原理如下。

    获取容器ID:

    sudo docker ps
    

    在此容器中启动bash(替换??????????上面步骤中的ID)

    sudo docker exec -it ??????? /bin/bash
    

    手动启动张紧轮板

    tensorboard --logdir=/tmp/logs
    

    我想在dockerfile的命令行中添加“tensorboard--logdir=/tmp/logs”,这样我可以删除手动步骤。我试过了

    CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/tf --ip 0.0.0.0 --no-browser --allow-root && tensorboard --logdir=/tmp/logs"]
    

    但它不起作用。所以本质上我的问题是如何在命令中运行2个东西。

    0 回复  |  直到 6 年前