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

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

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

Установите соответствующие драйверы NVIDIA на каждый компьютер на вашем сайте. Более подробно см. в Руководстве по установке NVIDIA CUDA для Linux.
1. Выполните действия после установки, чтобы настроить среду после установки CUDA Toolkit.
Установите nvidia-container-toolkit на этом компьютере, чтобы контейнеры блокнота могли использовать преимущества GPU. Обратитесь к руководству по установке NVIDIA Container Toollkit для скачивания ПО и документации для вашей ОС.
Запустите следующую команду на каждом компьютере, чтобы убедиться, что ваши элементы NVIDIA установлены правильно:

sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
Войдите на свой портал ArcGIS Enterprise как администратор и откройте ArcGIS Notebook Server Manager.
Откройте страницу Настройки и щелкните Runtimes.
Щелкните кнопку Изменить для среды выполнения, которую вы будете использовать в качестве основы среды выполнения с поддержкой ГП, либо ArcGIS Notebook Python 3 Advanced, либо ArcGIS Notebook Python 3 Standard. Скопируйте значение ID изображения. Щелкните Отмена для выхода из редактора.
На странице Рабочие среды щелкните Регистрация рабочей среды.
На странице Register Runtime укажите подходящее имя (например, GPU Runtime). Для значения ID изображения добавьте значение, скопированное на шаге 6.
Отметьте Включить графический процессор.
Щелкните Сохранить.
Убедитесь, что вы успешно настроили ArcGIS Notebook Server на использование NVIDIA GPU. Как участник портала с правами на Создание и редактирование блокнотов или Расширенные блокноты, если вы выбрали расширенную среду выполнения на шаге 6, создайте пустой блокнот. Когда вы выбираете рабочую среду блокнота, выберите вашу новую рабочую среду, готовую для ГП. Скопируйте в ячейку блокнота следующее и запустите ее.
```
import torch
torch.cuda.is_available()
```
Результат возвращается как True, поскольку для пакета torch.cuda необходимо, чтобы был запущен GPU.
Запустите следующую команду в новой ячейке, чтобы посмотреть конфигурацию ГПК вашего компьютера:

!nvidia-smi

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

Эта тема не относится к развертываниям Windows.

Эта тема не относится к развертываниям Linux.