Skip To Content

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 istniejące środowisko wykonawcze notatników w pliku Dockerfile, aby skonfigurować następujące zmienne środowiskowe, włączając opcję NVIDIAContainer Runtime dla kontenerów z akceleracją z użyciem procesorów graficznych.

    # Specify the existing notebook runtime imageId as FROM
    FROM <imageID>
    
    # 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 notatników.

    Wykonaj poniższe czynności, aby sprawdzić właściwość imageId istniejącego środowiska wykonawczego notatników. Zaloguj się do aplikacji ArcGIS Notebook Server Administrator Directory pod adresem https://notebookserver.domain.com:11443/arcgis/admin i przejdź do opcji notatniki > środowiska wykonawcze, wybierz środowisko wykonawcze, którego użyjesz jako podstawy środowiska wykonawczego z obsługą GPU, a następnie sprawdź, czy istnieje właściwość Image ID.

    1. Zaloguj się w aplikacji ArcGIS Notebook Server Administrator Directory pod adresem https://notebookserver.domain.com:11443/arcgis/admin.
    2. Kliknij opcję notatniki > środowiska wykonawcze.
    3. Wybierz środowisko wykonawcze, którego użyjesz jako podstawy środowiska wykonawczego z obsługą GPU i sprawdź, czy istnieje właściwość Image ID.
  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.

    Po zakończeniu budowania obrazu w komunikacie wynikowym zwrócona zostanie skrócona wartość właściwości imageId nowego obrazu.

  6. Pobierz pełną wartość właściwości imageId nowego kontenera.

    docker inspect <imageId>

  7. Zaloguj się w aplikacji ArcGIS Notebook Server Administrator Directory pod adresem https://notebookserver.domain.com:11443/arcgis/admin. Przejdź do opcji notatniki > środowiska wykonawcze i wybierz środowisko wykonawcze wybrane w etapie 4, a następnie kliknij opcję edytuj.
  8. Zaktualizuj ś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. Ustaw wartość dockerRuntime na nvidia. Zapisz zmiany.
  10. Sprawdź, czy aplikacja ArcGIS Notebook Server została pomyślnie skonfigurowana do użycia procesorów graficznych NVIDIA. Jeśli w etapie 4 wybrano zaawansowane środowisko wykonawcze, otwórz nowy notatnik jako członek portalu z uprawnieniami Tworzenie i edytowanie notatników lub Notatniki zaawansowane. 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

Aby usunąć 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.