使用Docker环境安装TensorFlow API

时间:2019-04-25 16:14:09

标签: docker tensorflow jupyter-notebook deeplab

我想使用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 repo克隆到tensorflow / models / research /中。

我在Internet上进行了一些研究以了解如何添加这些文件夹或在哪里找到tensorflow -docker- installation的位置,但没有结果。

所以请任何人帮助我。

谢谢

0 个答案:

没有答案