tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-15 16:52:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Move triton uninstall after uv sync in speedrun.sh for AMD

Browse Source
This commit is contained in:
google-labs-jules[bot] 2025-11-23 03:50:46 +00:00
parent 33b6b800fa
commit 994491b28d
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
speedrun.sh
View File

@ -32,11 +32,6 @@ if command -v nvidia-smi &> /dev/null; then
elif [ -e /dev/kfd ]; then elif [ -e /dev/kfd ]; then
echo "AMD GPU detected. Installing ROCm dependencies..." echo "AMD GPU detected. Installing ROCm dependencies..."
EXTRAS="amd" EXTRAS="amd"
# Explicitly uninstall triton if present, as it conflicts with pytorch-triton-rocm
# and can cause "ImportError: cannot import name 'Config' from 'triton'" errors
# if the NVIDIA version of triton (e.g. 3.4.0) is accidentally installed.
source .venv/bin/activate 2>/dev/null || true
uv pip uninstall -q triton || true
else else
echo "No dedicated GPU detected. Installing CPU dependencies..." echo "No dedicated GPU detected. Installing CPU dependencies..."
EXTRAS="cpu" EXTRAS="cpu"
@ -46,6 +41,13 @@ uv sync --extra $EXTRAS
# activate venv so that `python` uses the project's venv instead of system python # activate venv so that `python` uses the project's venv instead of system python
source .venv/bin/activate source .venv/bin/activate
# Explicitly uninstall triton if present, as it conflicts with pytorch-triton-rocm
# and can cause "ImportError: cannot import name 'Config' from 'triton'" errors
# if the NVIDIA version of triton (e.g. 3.4.0) is accidentally installed.
if [ "$EXTRAS" == "amd" ]; then
uv pip uninstall -q triton || true
fi
# ----------------------------------------------------------------------------- # -----------------------------------------------------------------------------
# wandb setup # wandb setup
# If you wish to use wandb for logging (it's nice!, recommended). # If you wish to use wandb for logging (it's nice!, recommended).