Skip To Content

Configurar ArcGIS Notebook Server para usar GPU

El uso de unidades de procesamiento gráfico (GPU) mejora rápidamente el tiempo de procesamiento de los modelos de aprendizaje de máquina. ArcGIS Notebook Server puede aprovechar las GPU de NVIDIA en su equipo host una vez que se han llevado a cabo algunos pasos adicionales.

Nota:

A partir de 10.8, los tiempos de ejecución de ArcGIS Notebook Server integrados incluyen Conda CUDA Toolkit para habilitar la compatibilidad con GPU. Anteriormente, este flujo de trabajo requería crear un tiempo de ejecución personalizado para incluir CUDA.

El siguiente flujo de trabajo tiene dos objetivos principales. El primero es instalar controladores y tiempo de ejecución de NVIDIA, lo que permitirá que el componente Docker de su sitio cree contenedores listos para GPU. El segundo es crear una copia del tiempo de ejecución Advanced del notebook que se configura para utilizar el tiempo de ejecución de NVIDIA. Todos los ArcGIS Notebooks abiertos con este tiempo de ejecución se iniciarán en contenedores listos para GPU. Aparte de eso, el nuevo tiempo de ejecución mantendrá todas las bibliotecas de Python del tiempo de ejecución Advanced del notebook.

Una vez que ArcGIS Notebook Server se haya instalado y configurado, siga estos pasos. Si su sitio de ArcGIS Notebook Server tiene varios equipos, siga los pasos del 1 al 3 en todos los equipos.

  1. Instale los controladores de NVIDIA adecuados en cada equipo de su sitio. Consulte el sitio web de NVIDIA para ver la información completa.
  2. Instale el tiempo de ejecución nvidia-docker 2.0 en el equipo para que los contenedores del notebook puedan hacer uso de las GPU. Consulte el Repositorio de NVIDIA-Docker en GitHub para ver las descargas y la documentación correspondiente a su sistema operativo en concreto.
  3. Ejecute el siguiente comando en cada equipo para asegurarse de que sus elementos de NVIDIA estén bien instalados:

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

  4. Inicie sesión en su portal de ArcGIS Enterprise como administrador y abra ArcGIS Notebook Server Manager.
  5. Abra la página Configuración y haga clic en Tiempos de ejecución.
  6. Haga clic en el icono Editar para el tiempo de ejecución etiquetado como ArcGIS Notebook Python 3 Advanced. Copie el valor dado por su Id. de imagen. Haga clic en Cancelar para salir del editor.
  7. Desde la página Tiempos de ejecución, haga clic en Registrar tiempo de ejecución.
  8. En la página Registrar tiempo de ejecución, proporcione un nombre adecuado (como Tiempo de ejecución de GPU) y asigne la versión como 10.8.1. En el valor Id. de imagen, agregue el valor que copió en el paso 5.
  9. Establezca el valor de Docker Runtime como nvidia. Haga clic en Registrar tiempo de ejecución para confirmar.
  10. Compruebe que ArcGIS Notebook Server se haya configurado correctamente para utilizar las GPU de NVIDIA. Como miembro del portal con el privilegio de Notebooks avanzados, cree un nuevo notebook en blanco. Cuando especifique el tiempo de ejecución del notebook, seleccione su nuevo tiempo de ejecución listo para GPU. Copie lo siguiente en una celda de notebook y ejecútela.

    import torch torch.cuda.is_available()
    La salida devuelve True, ya que el paquete de torch.cuda necesita GPU para ejecutarse.

  11. Ejecute el siguiente comando en una nueva celda para ver la configuración de GPU de su equipo:

    !nvidia-smi

Si desea eliminar la capacidad de que su sitio utilice GPU, abra la página Tiempos de ejecución en ArcGIS Notebook Server Manager y elimine el tiempo de ejecución creado en este flujo de trabajo.