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

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

После установки и настройки ArcGIS Notebook Server выполните следующие инструкции на том же компьютере.

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. Увеличьте время выполнения Advanced notebook в Dockerfile для установки следующих параметров среды, включив NVIDIA Container Runtime для контейнеров с ГПК-ускорением.

   # Specify the existing notebook runtime imageId as FROM FROM aa7a1a346e5b
   # Use RUN to issue a command to install
   # Declare environment variables with ENV
   #RUN conda install <your_preferred_gpu_package>
   ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
   ENV NVIDIA_REQUIRE_CUDA "cuda>=9.2"

   Вы также можете решить установить и другие модули Python, которые используют преимущества поддержки ГПК. Следуйте инструкциям в разделе Расширить среду выполнения блокнота для построения Dockerfile.
5. Постройте свой пользовательский образ контейнера с помощью следующего синтаксиса:

   docker build -t myGpuRuntime:v1.0 -f <path_to_Dockerfile> .

   Убедитесь, что указали период в конце команды.

6. После завершения построения образа появится итоговое сообщение с сокращенным imageId нового образа. Получите полное imageId своего нового контейнера.

   docker inspect <imageId>

7. Войдите в ArcGIS Notebook Server Administrator Directory на https://notebookserver.domain.com:11443/arcgis/admin. Перейдите к блокноты > среды выполнения и выберите среду выполнения Advanced, затем нажмите редактировать.
8. Обновите среду выполнения Advanced, чтобы использовать новый пользовательский образ. Замените существующее значение imageId значением своего образа, которое получили в шаге 6.
9. Установите значение dockerRuntime в значение nvidia. Сохраните внесенные изменения.
10. Убедитесь, что успешно настроили ArcGIS Notebook Server на использование ГПК NVIDIA. Как участник портала с правами на Advanced Notebooks, откройте новый блокнот. Скопируйте в ячейку следующее и запустите ее.

    import torch torch.cuda.is_available()

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

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

    !nvidia-smi

Даже если вы захотите удалить мощности вашего сайта ArcGIS Notebook Server, использующие ГПК, запустите операцию Restore Factory Runtimes в Administrator Directory. Она доступна по URL https://notebookserver.domain.com:11443/arcgis/admin/notebooks/runtimes/restore.