Использует ли sagemaker nvidia-docker или docker runtime == nvidia по умолчанию, или пользователь должен настроить вручную?

Question

Как указано в вопросе: «Использует ли sagemaker nvidia-docker или docker runtime == nvidia по умолчанию, или пользователю нужно вручную настроить?»

Некоторые распространенные сообщения об ошибках отображаются как «CannotStartContainerError. Пожалуйста, убедитесь, чтоконтейнер модели для варианта варианта-имени-1 запускается правильно, когда вызывается с помощью 'docker run serve'. "и он не показывается как работающий с драйвером nividia.

Итак, нам нужно настроить вручную?

Answer 1

Я использую изображение tenorflow-gpu в качестве базовых изображений для своих контейнеров, и я могу использовать gpu, не указывая ничего связанного с gpu.При создании Docker-контейнеров для Sagemaker вы должны остерегаться структуры папок и того, что ваш контейнер может запускаться с помощью команды serve (что указывает на ошибку).

Если у вас возникли проблемы с настройкой, я нахожу это пример самый полезный способ освоить его.