Я пытаюсь использовать тензор потока в докере, следуя инструкциям на
https://www.tensorflow.org/install/docker
У меня установлен докер 19.03 на машине Centos 7.7.1908, и я могу запустить nvidia-smi
docker run --gpus all nvidia/cuda:10.1-base nvidia-smi
Насколько я понимаю, использование nvidia-docker не рекомендуется.затем я загружаю tenorflow
docker pull tensorflow/tensorflow:latest-devel-py3
или
docker pull tensorflow/tensorflow:devel-gpu
, но когда я запускаю один из них
docker run -it --rm tensorflow/tensorflow:latest-devel-py3 python -c "import tensorflow as tf;"
, я получаю
Traceback (последний вызов был последним): Файл "", строка 1, в ModuleNotFoundError: Ни одному модулю с именем tenorflow не удалось изменить размер tty, используя размер по умолчанию
что мне не хватает?
хорошо, с тех пор я узнал, что
latest-gpu-py3
позволяет мне импортировать тензорный поток, тогда как
latest-devel-py3
- нет.Я бы подумал, что все эти изображения будут работать ...