Вопросы о докере, nvidia-docker, cudnn и тензорном потоке - PullRequest
1 голос
/ 30 мая 2019

У меня есть несколько вопросов по поводу образа докера cudnn и tenorflow-gpu. вот что я сделал на своем компьютере:

  1. Я установил docker-CE
  2. Я установил nvidia-docker
  3. Я вытащил tensorflow:latest-gpu-py3-jupyter изображение из докера.
  4. Я создаю один контейнер с помощью следующей команды:

    sudo docker create -ti --rumtime=nvidia -p 21001:22 -v
    /home/project:/project tensorflow:latest-gpu-py3-jupyter /bin/bash
    
  5. Я запускаю и забираюсь в контейнер, затем тренирую свою модель, которая построена с использованием Keras. но я получаю одно сообщение об ошибке --- 'Ошибка сегментации (ядро сброшено)'

Кажется, что cudnn не установлен в моем контейнере. Кто-нибудь знает, почему это произошло? Не содержит ли изображение tenorflow библиотеку cudnn?

...