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