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

Merge pull request #16 from LokiMetaSmith/fix-amd-install

Browse Source 
Fix AMD Triton conflict in speedrun.sh
This commit is contained in:
Lawrence R Kincheloe III 2025-11-22 21:40:05 -06:00 committed by GitHub
commit 33b6b800fa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
speedrun.sh
View File

@ -32,6 +32,11 @@ if command -v nvidia-smi &> /dev/null; then
elif [ -e /dev/kfd ]; then
echo "AMD GPU detected. Installing ROCm dependencies..."
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
echo "No dedicated GPU detected. Installing CPU dependencies..."
EXTRAS="cpu"