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.
- Zainstaluj odpowiednie sterowniki NVIDIA na komputerze. Wszystkie informacje znajdziesz w witrynie NVIDIA.
- 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.
- 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
- 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.
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.# 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"
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.
- Zaloguj się w aplikacji ArcGIS Notebook Server Administrator Directory pod adresem https://notebookserver.domain.com:11443/arcgis/admin.
- Kliknij opcję notatniki > środowiska wykonawcze.
- Wybierz środowisko wykonawcze, którego użyjesz jako podstawy środowiska wykonawczego z obsługą GPU i sprawdź, czy istnieje właściwość Image ID.
- Zbuduj niestandardowy obraz kontenera przy użyciu polecenia o następującej składni:
Pamiętaj, aby postawić kropkę na końcu polecenia.docker build -t myGpuRuntime:v1.0 -f <path_to_Dockerfile> .
Po zakończeniu budowania obrazu w komunikacie wynikowym zwrócona zostanie skrócona wartość właściwości imageId nowego obrazu.
- Pobierz pełną wartość właściwości imageId nowego kontenera.
docker inspect <imageId>
- 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.
- 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.
- Ustaw wartość dockerRuntime na nvidia. Zapisz zmiany.
- 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ą.
Powinny zostać zwrócone dane wynikowe w postaci True, ponieważ do uruchomienia pakietu torch.cuda wymagane są procesory graficzne.import torch torch.cuda.is_available()
- 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.