Настройка ArcGIS Notebook Server для использования ГПК

Использование мощностей графических процессоров (ГПК) существенно улучшает время обработки для моделей машинного обучения. ArcGIS Notebook Server может использовать преимущества ГПК NVIDIA на своем компьютере хоста, но для этого необходимо выполнить ряд настроек.

Примечание:

Начиная с версии 10.8, встроенные ArcGIS Notebook Server runtimes включают Conda CUDA Toolkit для поддержки графического процессора ГПК. Ранее этот рабочий процесс требовал создания настраиваемой runtime для включения CUDA.

Следующий рабочий процесс имеет две основные цели. Первый заключается в установке драйверов NVIDIA и runtime, которые позволят вашему компоненту сайта Docker построение ГПК-готовых контейнеров. Второе – создать копию Advanced notebook runtime, настроенную для использования runtime NVIDIA. Все ArcGIS Notebooks, открытые с помощью этой runtime, будeт запущены в ГПК-готовых контейнерах. Кроме того, новая рабочая среда сохранит все библиотеки Python для Advanced notebook runtime.

После установки и настройки ArcGIS Notebook Server выполните следующие инструкции. Если на вашем сайте ArcGIS Notebook Server несколько компьютеров, выполните шаги с 1 по 3 на всех компьютерах.

1. Установите соответствующие драйверы NVIDIA на каждую машину на вашем сайте. Подробнее см. Веб-сайт NVIDIA .
2. Установите среду выполнения nvidia-docker 2.0 на этом компьютере, чтобы контейнеры ноутбука могли использовать преимущества ГПК. Посмотрите NVIDIA-Docker repository on GitHub для скачивания и документации для вашей конкретной ОС.
3. Запустите следующую команду на каждом компьютере, чтобы убедиться, что ваши элементы NVIDIA установлены правильно:

   docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

4. Войдите на ваш портал ArcGIS Enterprise с правами администратора и откройтеArcGIS Notebook Server Manager.
5. Откройте страницу Настройки и щелкните Runtimes.
6. Щелкните значок Редактировать для runtime, подписанной ArcGIS Notebook Python 3 Advanced. Скопируйте значение, указанное для его ID изображения. Щелкните Отмена для выхода из редактора.
7. На странице Runtimes щелкните Register Runtime.
8. На странице Register Runtime укажите подходящее имя (например, GPU Runtime) и укажите версию как 10.9.1. Для значения ID изображения добавьте значение, скопированное на шаге 5.
9. Установите значение Docker Runtime равным nvidia. Щелкните Register Runtime, чтобы применить изменения.
10. Убедитесь, что успешно настроили ArcGIS Notebook Server на использование ГПК NVIDIA. Как участник портала с правами на Advanced Notebooks, создайте новый пустой блокнот. Когда вы задаете рабочую среду блокнота, выберите вашу новую рабочую среду ГПК. Скопируйте в ячейку блокнота следующее и запустите ее.

    import torch torch.cuda.is_available()

    Результат возвращается как True, поскольку для пакета torch.cuda необходимо, чтобы был запущен ГПК.

11. Запустите следующую команду в новой ячейке, чтобы посмотреть конфигурацию ГПК вашего компьютера:

    !nvidia-smi

Если вы хотите удалить емкость для своего сайта для использования ГПК, откройте страницу Runtimes в ArcGIS Notebook Server Manager и удалите рабочую среду, созданную в этом рабочем процессе.