pip install aitraining torch torchvision torchaudio

uv pip install aitraining torch torchvision torchaudio

aitraining --version

aitraining

import torch
print(torch.cuda.is_available())  # Should print True
print(torch.cuda.device_count())  # Number of GPUs

import torch
print(torch.backends.mps.is_available())  # Should print True

export HF_TOKEN="your_token_here"
export WANDB_API_KEY="your_key_here"