Skip To Content

Настройка ArcGIS Notebook Server для использования графических процессоров

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

Примечание:

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

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

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

  1. Установите соответствующие драйверы NVIDIA на каждую машину на вашем сайте. Полную информацию см. на веб-сайте NVIDIA.
  2. Установите среду выполнения nvidia-docker 2.0 на этом компьютере, чтобы контейнеры блокнота могли использовать преимущества ГП. Обратитесь к репозиторию NVIDIA-Docker на 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. Щелкните кнопку Изменить для среды выполнения, которую вы будете использовать в качестве основы среды выполнения с поддержкой ГП, либо ArcGIS Notebook Python 3 Advanced, либо ArcGIS Notebook Python 3 Standard. Скопируйте значение ID изображения. Щелкните Отмена для выхода из редактора.
  7. На странице Рабочие среды щелкните Регистрация рабочей среды.
  8. На странице Регистрация рабочей среды укажите подходящее имя (например, Рабочая среда с ГП) и укажите версию как 11.2. Для значения ID изображения добавьте значение, скопированное на шаге 5.
  9. Установите значение Рабочая среда Docker равным nvidia. Щелкните Регистрация рабочей среды для подтверждения.
  10. Убедитесь, что вы успешно настроили ArcGIS Notebook Server на использование ГП NVIDIA. Как участник портала с правами на Создание и редактирование блокнотов или Расширенные блокноты, если вы выбрали расширенную среду выполнения на шаге 6, создайте пустой блокнот. Когда вы выбираете рабочую среду блокнота, выберите вашу новую рабочую среду, готовую для ГП. Скопируйте в ячейку блокнота следующее и запустите ее.

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

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

    !nvidia-smi

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