Pular para o conteúdo principal
Última atualização em

requirements.yaml referência

info

Beta

O AI Runtime CLI está em Beta.

requirements.yaml define as bibliotecas Python instaladas para sua carga de trabalho. É referenciado a partir do bloco environment do YAML da carga de trabalho:

YAML
environment:
  dependencies: requirements.yaml

O formato do arquivo segue a Especificação do Ambiente Base do Databricks.

Exemplo​

YAML
environment_version: '4'
dependencies:
  - --index-url https://pypi.org/simple
  - -r '/Workspace/Shared/requirements.txt'
  - my-library==6.1
  - /Workspace/Shared/Path/To/simplejson-3.19.3-py3-none-any.whl

No exemplo acima:

  1. A imagem do Ambiente Base v4 é utilizada.
  2. Um requirements.txt referenciado no workspace está instalado.
  3. my-library está instalada na versão 6.1.
  4. Uma wheel no workspace é instalada.

Reutilizar um existente requirements.txt​

Caso já se mantenha um requirements.txt, é possível referenciá-lo diretamente de seu workspace:

YAML
version: '4'
dependencies:
  - -r '/Workspace/path/to/requirements.txt'

O campo version é opcional e assume "4" como padrão se não especificado.

Exemplo de Axolotl​

Arquivo de dependência para treinamento com Axolotl:

YAML
version: '4'
dependencies:
  # Standard build tools and MLflow
  - packaging
  - setuptools
  - wheel
  - ninja
  - mlflow>=3.6

  # Axolotl with flash-attn and deepspeed (no build isolation)
  - --no-build-isolation
  - axolotl[flash-attn,deepspeed]

  # Flash attention wheel (no deps — specific CUDA / torch version)
  - --no-deps
  - https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl
Nesta página