Konfiguracja aplikacji ArcGIS Notebook Server na potrzeby korzystania z procesorów graficznych

Użycie procesorów graficznych (jednostek GPU) skutecznie poprawia czas przetwarzania modeli samouczenia się maszyn. Aplikacja ArcGIS Notebook Server będzie mogła korzystać z procesorów graficznych NVIDIA na komputerze hosta, jeśli zostanie wykonanych kilka dodatkowych czynności.

Po zainstalowaniu i skonfigurowaniu aplikacji ArcGIS Notebook Server wykonaj następujące czynności przy użyciu tego samego komputera.

1. Zainstaluj odpowiednie sterowniki NVIDIA na komputerze. Wszystkie informacje znajdziesz w witrynie NVIDIA.
2. Zainstaluj środowisko wykonawcze nvidia-docker 2.0 na komputerze, aby kontenery notatnika mogły korzystać z procesorów graficznych. W repozytorium NVIDIA-Docker w serwisie GitHub można znaleźć materiały do pobrania i dokumentację dotyczącą konkretnego systemu operacyjnego.
3. Uruchom następujące polecenie, aby upewnić się, że elementy NVIDIA są poprawnie zainstalowane:

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

4. Rozszerz środowisko wykonawcze Notatnik zaawansowany w pliku Dockerfile, aby skonfigurować następujące zmienne środowiskowe, włączając opcję NVIDIA Container Runtime dla kontenerów z akceleracją z użyciem procesorów graficznych.

   # 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"

   Możesz także zdecydować się na zainstalowanie innych modułów Python, które obsługują procesory graficzne. Aby zbudować plik Dockerfile, wykonaj czynności opisane w temacie Rozszerzanie środowiska wykonawczego notatnika.
5. Zbuduj niestandardowy obraz kontenera przy użyciu polecenia o następującej składni:

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

   Pamiętaj, aby postawić kropkę na końcu polecenia.

6. Po zakończeniu budowania obrazu w komunikacie wynikowym zwrócona zostanie skrócona wartość identyfikatora imageId nowego obrazu. Pobierz pełną wartość identyfikatora imageId nowego kontenera.

   docker inspect <imageId>

7. Zaloguj się w aplikacji ArcGIS Notebook Server Administrator Directory w lokalizacji https://notebookserver.domain.com:11443/arcgis/admin. Przejdź do opcji Notatniki > Środowiska wykonawcze i wybierz zaawansowane środowisko wykonawcze. Następnie kliknij przycisk Edytuj.
8. Zaktualizuj zaawansowane środowisko wykonawcze, aby używało nowego niestandardowego obrazu. Zastąp istniejącą wartość imageId wartością obrazu, która została podana w etapie 6.
9. Skonfiguruj parametr dockerRuntime na wartość nvidia. Zapisz zmiany.
10. Sprawdź, czy aplikacja ArcGIS Notebook Server została pomyślnie skonfigurowana do użycia procesorów graficznych NVIDIA. Otwórz nowy notatnik jako członek portalu z uprawnieniami Zaawansowane notatniki. Skopiuj poniższą treść do komórki i uruchom ją.

    import torch torch.cuda.is_available()

    Powinny zostać zwrócone dane wynikowe w postaci True, ponieważ do uruchomienia pakietu torch.cuda wymagane są procesory graficzne.

11. Uruchom następujące polecenie w nowej komórce, aby wyświetlić konfigurację procesorów graficznych używanego komputera:

    !nvidia-smi

Jeśli kiedykolwiek zajdzie potrzeba usunięcia możliwości korzystania z procesorów graficznych w witrynie aplikacji ArcGIS Notebook Server, uruchom operację Przywróć fabryczne środowiska wykonawcze w aplikacji Administrator Directory. Jest ona dostępna pod adresem URL https://notebookserver.domain.com:11443/arcgis/admin/notebooks/runtimes/restore.