Pular para o conteúdo
Skillpedia
Deploy e nuvem (DevOps)41 mil · #4852

spark-environment-setup

Faz seu agente configurar um ambiente para treinar modelos de inteligência artificial em hardware específico.

O que ela faz

Esta ferramenta ajuda a configurar seu computador (DGX Spark) para rodar programas de inteligência artificial que usam gráficos (como PyTorch). Ela garante que as partes certas do programa funcionem juntas corretamente.

Quando usar

  • ✓Ao instalar programas de IA pela primeira vez no DGX Spark.
  • ✓Quando um programa de IA dá erro ao iniciar ou usar a placa de vídeo.
  • ✓Ao decidir se usa um programa pronto (container) ou instala tudo manualmente.

Exemplo do que pedir para a IA

“Como configuro meu DGX Spark para rodar PyTorch sem erros?”

Quer saber mais? Abra as seções abaixo.

Como usar

O jeito mais fácil: copie a mensagem abaixo e cole no Claude, ChatGPT, Cursor ou outra IA. Ela lê a skill e segue as instruções.

Link da skill

https://github.com/wshobson/agents/blob/main/plugins/dgx-spark-ops/skills/spark-environment-setup/SKILL.md

Mensagem pronta para colar

Use esta skill para me ajudar: https://github.com/wshobson/agents/blob/main/plugins/dgx-spark-ops/skills/spark-environment-setup/SKILL.md Leia o arquivo SKILL.md desse link e siga as instruções dele na minha tarefa.
Como funciona

Ela verifica se os programas (como PyTorch, Unsloth) são compatíveis com a versão do CUDA (o sistema da placa de vídeo) do seu DGX Spark. Ela recomenda usar um 'container' (pacote pronto) ou seguir uma lista exata de comandos para instalar tudo.

Instalação

Jeito mais fácil: copie o link da skill e cole no Claude, ChatGPT ou Cursor pedindo para seguir as instruções.

No terminal (Claude Code, Codex, Cursor), na pasta do projeto:

npx skills add wshobson/agents --skill spark-environment-setup

Ou baixe só o arquivo:

curl -L https://raw.githubusercontent.com/wshobson/agents/main/plugins/dgx-spark-ops/skills/spark-environment-setup/SKILL.md -o SKILL.md
Conteúdo técnico (SKILL.md)

Tradução do texto original do autor no GitHub. Os comandos e códigos estão como no original.

Configuração do Ambiente Spark

O DGX Spark vem com um chip GB10 Grace Blackwell: CPU aarch64, GPU SM121, 128GB de memória unificada, CUDA 13. Esta é uma plataforma mais restrita e nova do que uma caixa CUDA 12 x86 padrão, então a seleção de pacotes e a compatibilidade de ABI importam mais do que o usual — o ecossistema de wheels para aarch64 + CUDA 13 ainda está em desenvolvimento.

Quando Usar Esta Skill

  • Configurando uma nova caixa Spark para treinamento ou inferência.
  • Encontrando um erro de importação mencionando libcudart, um símbolo ausente, ou uma wheel que "foi instalada com sucesso, mas não carrega".
  • Uma instalação de framework (PyTorch, Unsloth, TRL, vLLM, xformers) falha, trava ou volta silenciosamente para a CPU.
  • Decidindo se deve usar um contêiner NGC ou pip puro.
  • Restaurando uma configuração funcional após uma reinstalação do sistema operacional ou uma atualização da imagem base, precisando verificar novamente do zero.

Cada um desses cenários aceita a mesma correção geral: combinar o contêiner/wheel com CUDA 13 e SM121, não lute contra a ABI.

Regra do Contêiner em Primeiro Lugar

Decisão rápida, antes dos detalhes abaixo:

  • Trabalho padrão de treinamento/inferência → Contêiner PyTorch NGC.
  • Fine-tuning focado em Unsloth → Contêiner Unsloth (ele já vem com a combinação de Triton/xformers/transformers fixada e validada para esse caminho).
  • Nenhum dos dois se encaixa (pacote de sistema personalizado, interpretador de IDE local) → pip puro, seguindo a sequência exata mais adiante.

Prefira um contêiner. Use nvcr.io/nvidia/pytorch:25.09-py3 como base para trabalho geral — a tag mais recente confirmada como funcional neste hardware; baixe uma tag mais nova e aprovada se disponível localmente em vez de bloquear em 25.11-py3. A tag do NGC está datada, então executá-la diretamente é bom:

docker run --runtime=nvidia --gpus all -it --rm \
  nvcr.io/nvidia/pytorch:25.09-py3

unsloth/unsloth:dgxspark-latest é uma tag em movimento em contraste — resolva e fixe seu digest antes de executá-la para qualquer coisa reproduzível; a tag pura é apenas uma etapa de descoberta, não a invocação padrão. Sequência completa de pull-inspect-pin e justificativa de flags/montagens de volume para diretórios de execução de finetuning/: references/container-workflow.md. Trate o pip puro como a exceção.

A razão para a abordagem de contêiner em primeiro lugar é a fixação (pinning), não a conveniência. As versões de Triton, xformers e transformers interagem de forma restrita com o alvo SM121 do GB10 e CUDA 13; um contêiner trava todos eles juntos contra uma combinação já validada neste hardware. O pip puro deixa essa resolução para você, um erro de importação de cada vez.

Quando o pip puro for justificado, siga a sequência de instalação do playbook da NVIDIA literalmente e em ordem:

pip install "transformers==5.13.1" "peft==0.19.1" "hf_transfer==0.1.9" "datasets==4.3.0" "trl==1.8.0"
pip install --no-deps "unsloth==2026.7.2" "unsloth_zoo==2026.7.2" "bitsandbytes==0.49.2"
pip install -U "torchao==0.17.0"

O flag --no-deps do segundo comando não é opcional — permitir que o pip resolva novamente a árvore de dependências do Unsloth em aarch64 é uma maneira comum de puxar uma build incompatível de torch ou triton. A terceira linha também não é opcional: a imagem base do NGC empacotada com torchao é muito antiga para o caminho LoRA-attach dos peft atuais (ImportError: ... torchao ... only versions above 0.16.0 are supported) — a hard blocker, not a warning. Every == o pin acima é crucial, retirado da matriz de versão conhecida e funcional datada em references/stack-matrix.md (a data de Last verified governa a obsolescência) — uma instalação sem pin resolve as versões atuais do PyPI bem fora do que esta versão do Unsloth suporta.

Baixe uma tag nova quando um novo lançamento aprovado for anunciado. Reconstrua localmente a partir de uma das duas bases apenas quando um projeto precisar de um pacote de sistema extra — não para "atualizar" um componente que a imagem já fixa. Detalhes sobre ambos os caminhos: references/container-workflow.md.

Mais um pré-voo: playbooks oficiais do DGX Spark já foram lançados com erros. Verifique issues recentes em github.com/NVIDIA/dgx-spark-playbooks (e os outros recursos em references/stack-matrix.md) antes de confiar em uma receita literalmente para uma execução longa.

A Regra da ABI

A falha mais comum no Spark é uma incompatibilidade de ABI CUDA 12/13: uma wheel construída contra libcudart.so.12 carregada em um sistema que só tem libcudart.so.13. A instalação geralmente é bem-sucedida; a falha surge mais tarde como um erro de símbolo ausente ou um segfault que não aponta claramente para CUDA.

Correção: baixe wheels de download.pytorch.org/whl/cu130 (as builds aarch64 com tag cu130), ou use um dos contêineres acima, que já carregam uma build compatível. Antes de perseguir um

…

Perguntas frequentes

O que é um 'container'?

É como uma caixa pronta com todos os programas e configurações necessárias para rodar algo específico, evitando problemas de compatibilidade.

O que é 'ABI mismatch'?

É quando duas partes do programa esperam versões diferentes de uma biblioteca fundamental (como a do CUDA), causando erros.

Por que usar um container é melhor?

Porque garante que todas as peças (PyTorch, CUDA, etc.) foram testadas juntas e funcionam bem no seu DGX Spark, evitando dores de cabeça com compatibilidade.

Detalhes

Descrição original do autor

Set up a working ML training/inference environment on NVIDIA DGX Spark (GB10, aarch64, CUDA 13). Use when installing PyTorch/Unsloth/TRL/vLLM on DGX Spark, hitting libcudart or wheel-ABI errors on aarch64, or choosing between NGC containers and bare pip installs.

repo: wshobson/agents

Popularidade

Estrelas no GitHub
41 mil
Posição no ranking
#4852
Licença
MIT
Atualizada em
15/10/2026

Essa skill é útil?

0

Entre para votar.

Comentários

  • …

Mais skills deste repositório

Skills parecidas