Установите API TensorFlow, используя среду Docker - PullRequest
Новая
       11

Установите API TensorFlow, используя среду Docker

3 голосов
/ 25 апреля 2019

Я хочу обучить модель с использованием 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-репо в tenorflow / models / research /.

Я провожу некоторые исследования в Интернете, чтобы узнать, как добавить эти папки или где мы находим местоположение установки tenorflow -docker-, но безрезультатно.

Так что, пожалуйста, кто-нибудь может мне помочь.

Thankyou

...