我想用Tensorflow训练一个模型,所以为了避免python、CUDA、Tensorflow之间的依赖问题......,我决定使用Docker。
要安装 Tensorflow,我使用了这些命令
# install nvidia docker
docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
# install tensorflow docker
docker pull tensorflow/tensorflow:1.12.0-gpu-py3
# run
docker run --runtime=nvidia -it --rm -v /$(pwd)/Desktop/DeepLab:/notebooks -p 8888:8888 tensorflow/tensorflow:1.12.0-gpu-py3
现在,我想安装 DeepLab 模型,为此我必须将 git 存储库克隆到 tensorflow/models/research/ 中。
我在互联网上进行了一些研究,以了解如何添加这些文件夹或我们在哪里找到 tensorflow -docker- 安装的位置,但没有结果。
所以拜托,谁能帮助我。
谢谢