Fix GPU-less CPU use on Linux with specific Torch indexes

2025-10-21 19:52:21 +00:00
parent defd1246aa
commit 901b075605
3 changed files with 29 additions and 13 deletions
--- a/dev/runcpu.sh
+++ b/dev/runcpu.sh
@@ -14,7 +14,7 @@ NANOCHAT_BASE_DIR="$HOME/.cache/nanochat"
 mkdir -p $NANOCHAT_BASE_DIR
 command -v uv &> /dev/null || curl -LsSf https://astral.sh/uv/install.sh | sh
 [ -d ".venv" ] || uv venv
-uv sync
+uv sync --extra cpu
 source .venv/bin/activate
 if [ -z "$WANDB_RUN" ]; then
    WANDB_RUN=dummy