Sdílet prostřednictvím


AI prostředí Databricks

Prostředí Databricks AI je kurátorovaný modul runtime s podporou GPU přizpůsobený vývoji umělé inteligence. Podporuje se v bezserverovém prostředí GPU 4 a novějším.

Toto nové prostředí zjednodušuje vývoj tím, že poskytuje plně nakonfigurovaný zásobník knihoven pro strojové učení (včetně architektur, jako jsou PyTorch, HuggingFace Transformers atd.) a nativní podpora gpu. Integruje se s poznámkovými bloky Databricks, katalogem Unity a MLflow a poskytuje integrované prostředí. S prostředím AI můžou týmy ML jednoduše vybrat cluster GPU připravený k provozu a okamžitě začít trénovat modely, a ne strávit dny při nastavování a řešení potíží.

Připojení k prostředí AI

Použití AI prostředí Databricks z poznámkového bloku Databricks připojeného k bezserverovým GPU výpočetním prostředkům:

  1. V poznámkovém bloku klikněte nahoře na rozevírací nabídku Připojit a vyberte Serverless GPU.
  2. Kliknutím na ikonu Prostředí otevřete boční panel Prostředí .
  3. V poli Akcelerátor vyberte A10.
  4. V poli Základní prostředí vyberte AI v4 pro prostředí AI.
  5. Pokud jste v poli Základní prostředí zvolili Možnost Žádné, vyberte verzi prostředí.
  6. Klikněte na Použít a pak potvrďte, že chcete ve svém prostředí notebooku použít bezserverové GPU výpočetní prostředky.

Nastavení prostředí Databricks AI pro notebookovou úlohu na serverless GPU:

  1. V nastavení úlohy poznámkového bloku klikněte v části Prostředí a knihovny na Upravit prostředí poznámkového bloku.
  2. Kliknutím na ikonu Prostředí otevřete boční panel Prostředí .
  3. V poli Základní prostředí vyberte AI v4 pro prostředí AI.

Nové běhy úloh budou schopny integrovat prostředí Databricks AI.

časté otázky

Jaký je rozdíl mezi prostředím Databricks AI a prostředím v Databricks Runtime pro Machine Learning?

Databricks Runtime pro Machine Learning i prostředí AI Databricks poskytují předem nakonfigurované výpočetní prostředí přizpůsobené případům použití AI/ML. I když se Databricks Runtime pro Machine Learning používá s klasickými výpočetními prostředky, prostředí Databricks AI je určené pro výpočetní prostředí GPU bez serveru.

Prostředí v prostředí Databricks Runtime pro Machine Learning i prostředí AI Databricks zahrnuje běžné balíčky strojového učení s některými rozdíly. Především prostředí Databricks AI obsahuje aktualizační balíčky, ale nezahrnuje Tensorflow a GraphFrames. Další informace o tom, co je součástí prostředí Databricks AI a Databricks Runtime pro Machine Learning, najdete v poznámkách k verzi bezserverového prostředí a poznámkách k verzi Modulu runtime Databricks.

Známé problémy

  • Prostředí AI Databricks nefunguje s polem Prostředí a knihovny v konfiguraci úlohy poznámkového bloku. Pokud z daného pole vytvoříte nové prostředí úloh, možná nebudete moct vybrat prostředí Databricks AI.
  • Prostředí Databricks AI nepodporuje export prostředí.