Gunakan keterampilan agen dengan AI Runtime

databricks-ai-runtime keterampilan agen memberikan instruksi kepada agen pengkode AI, seperti Claude Code, Cursor, dan Codex, untuk bekerja dengan AI Runtime melalui databricks air perintah Databricks CLI. Dengan keterampilan terpasang, Anda dapat meminta agen Anda menulis workload YAML, mengirimkan proses pelatihan GPU, dan memeriksa kemajuannya.

Requirements

Instal keterampilan

Jalankan perintah berikut:

databricks aitools install --skills-only --skills databricks-ai-runtime

Ikuti petunjuk untuk memilih agen AI dan cakupan instalasi Anda. Untuk opsi instalasi, lihat aitools grup perintah.

Apa yang dicakup oleh keterampilan ini

Capability Apa yang dapat dilakukan agen
Beban kerja penulisan Ubah permintaan menjadi workload YAML dengan komputasi, lingkungan, sumber kode, dan perintah peluncuran untuk framework Anda, seperti torchrun, Accelerate, atau DeepSpeed.
Konfigurasikan eksekusi Atur variabel lingkungan, rahasia, hiperparameter, retry, dan izin.
Kirim proses Validasi konfigurasi dengan --dry-run, lalu kirim dengan databricks air run dan kunci idempotensi.
Periksa dan pantau eksekusi Dapatkan status dengan databricks air get dan databricks air list, log stream dengan databricks air logs, dan arahkan Anda ke MLflow run untuk metrik.
Batalkan proses Batalkan satu, beberapa, atau semua run aktif Anda dengan databricks air cancel.
Kegagalan penelusuran kesalahan Ambil log dari run yang gagal dan tampilkan error.
Menggunakan gambar Docker kustom Buat dan unggah image kustom, lalu referensikan image tersebut pada workload.

Sumber daya tambahan