2017-10-21 42 views
1

我正在使用docker來運行tensorflow和再培訓啓動模塊。我使用下面的代碼:Docker的啓動和工作目錄

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的目錄應該是搬運工容器內。但是,我希望此目錄位於我的主目錄中(/ user/documents/flower_photos)。我該怎麼辦?

回答

1

你可以use a volume以主機文件夾到一個容器文件夾關聯:

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

這樣一來,開始模塊會找到你的主機內容的現有文件夾。