Databricks Runtime 12.1 para Machine Learning (EoS)

O Databricks Runtime 12.1 para Aprendizagem Automática fornece um ambiente pronto para uso imediato para aprendizagem automática e ciência de dados baseado no Databricks Runtime 12.1 (EoS). O Databricks Runtime ML contém muitas bibliotecas populares de aprendizado de máquina, incluindo TensorFlow, PyTorch e XGBoost. O Databricks Runtime ML inclui o AutoML, uma ferramenta para treinar automaticamente pipelines de aprendizado de máquina. O Databricks Runtime ML também suporta treinamento distribuído de aprendizado profundo usando o Horovod.

Para obter mais informações, incluindo instruções para criar um cluster de ML do Databricks Runtime, consulte IA e aprendizado de máquina no Databricks.

Novos recursos e melhorias

O Databricks Runtime 12.1 ML é construído sobre o Databricks Runtime 12.1. Para obter informações sobre o que há de novo no Databricks Runtime 12.1, incluindo Apache Spark MLlib e SparkR, consulte as notas de versão do Databricks Runtime 12.1 (EoS ).

AutoML

A partir do Databricks Runtime 12.1 ML, a API Python AutoML permite especificar um nome personalizado para o experimento gerado pelo AutoML. Use o parâmetro experiment_name.

Para obter mais informações sobre AutoML, consulte O que é AutoML?.

Ambiente do sistema

O ambiente do sistema no Databricks Runtime 12.1 ML difere do Databricks Runtime 12.1 da seguinte maneira:

• DBUtils: o Databricks Runtime ML não inclui o utilitário de biblioteca (dbutils.library) (legado). Em vez disso, use comandos %pip. Ver Bibliotecas de Python com âmbito de notebook.
• Para clusters de GPU, o Databricks Runtime ML inclui as seguintes bibliotecas de GPU NVIDIA:
  • CUDA 11,3
  • cuDNN 8.0.5.39
  • NCCL 2.9.9
  • TensorRT 7.2.2

O Databricks Runtime 12.1 ML inclui XGBoost 1.7.2, que não suporta clusters GPU com capacidade de computação 5.2 e inferior.

Bibliotecas

As seções a seguir listam as bibliotecas incluídas no Databricks Runtime 12.1 ML que diferem daquelas incluídas no Databricks Runtime 12.1.

Nesta secção:

• Bibliotecas de nível superior
• Python bibliotecas
• Bibliotecas R
• Bibliotecas Java e Scala (cluster Scala 2.12)

Bibliotecas de nível superior

O Databricks Runtime 12.1 ML inclui as seguintes bibliotecas de camada superior:

• GraphFrames
• Horovod e HorovodRunner
• MLflow
• PyTorch
• Spark-TensorFlow Connector
• TensorFlow
• TensorBoard
• Scikit-learn

Bibliotecas Python

O Databricks Runtime 12.1 ML utiliza o Virtualenv para gestão de pacotes em Python e inclui muitos pacotes populares de ML.

Além dos pacotes especificados nas seções a seguir, o Databricks Runtime 12.1 ML também inclui os seguintes pacotes:

• hiperopt 0.2.7.db1
• Sparkdl 2.3.0-DB3
• AutoML 1.15.0 |

Para reproduzir o ambiente Databricks Runtime ML Python no seu ambiente virtual Python local, descarregue o ficheiro requirements-12.1.txt e execute pip install -r requirements-12.1.txt. Este comando instala todas as bibliotecas open source que o Databricks Runtime ML utiliza, mas não instala as bibliotecas desenvolvidas pelo Databricks, como databricks-automl, databricks-feature-store, ou o fork Databricks de hyperopt.

Bibliotecas Python em clusters de CPU

Bibliotecas Python em clusters de GPU

Bibliotecas R

As bibliotecas R são idênticas às bibliotecas R no Databricks Runtime 12.1.

Bibliotecas Java e Scala (cluster Scala 2.12)

Para além das bibliotecas Java e Scala no Databricks Runtime 12.1, o Databricks Runtime 12.1 ML contém os seguintes JARs:

Clusters de CPU

Clusters de GPU