我想使用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的位置,但没有结果。
所以请任何人帮助我。
谢谢