Docker的Inception和Work目录

时间:2017-10-21 20:20:20

标签: docker tensorflow

我正在使用docker来运行tensorflow和retrain inception模块。我使用以下代码:

docker run -it \
--publish 6006:6006 \
--volume ${HOME}/tf_files:/tf_files \
--workdir /tf_files \
tensorflow/tensorflow:1.1.0 bash

然后我用

python retrain.py
bottleneck_dir=bottlenecks
how_many_training_steps=500
model_dir=inception
summaries_dir=training_summaries/basic
output_graph=retrained_graph.pb
output_labels=retrained_labels.txt
image_dir=flower_photos

当我运行这些代码时,flower_photos的目录应该在docker容器内。但是,我希望此目录位于我的主目录中(/ user / documents / flower_photos)。我该怎么办?

1 个答案:

答案 0 :(得分:1)

您可以use a volume将主机文件夹与容器文件夹相关联:

docker run -it \
...
-v /user/documents/flower_photos:/path/to/inception/flower_photos

这样,初始模块就会找到包含主机内容的现有文件夹。