setup.sh

#!/usr/bin/env bash
set -euo pipefail

ENV_NAME="moAI"
YAML="config/conda_env.yaml"

conda env remove -n "$ENV_NAME" -y || true
conda env create -f "$YAML" -n "$ENV_NAME"
conda run -n "$ENV_NAME" conda remove -y pytorch torchvision cudatoolkit || true

conda run -n "$ENV_NAME" python -m pip install --upgrade pip
conda run -n "$ENV_NAME" python -m pip install torch torchvision
conda run -n "$ENV_NAME" python -m pip install -e megapose6d/

conda run -n "$ENV_NAME" --no-capture-output \
  python -c 'import torch, torchvision, sys, platform; \
print("Python:", sys.version.split()[0], platform.platform()); \
print("Torch:", torch.__version__, "CUDA:", torch.version.cuda, "CUDA avail:", torch.cuda.is_available()); \
print("Torchvision:", torchvision.__version__)'