4

我想用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- 安装的位置,但没有结果。

所以拜托,谁能帮助我。

谢谢

4

0 回答 0