Skip to content

Instantly share code, notes, and snippets.

View morgangiraud's full-sized avatar
🚤
Drowning under all the innovations flooding ML

Morgan Giraud morgangiraud

🚤
Drowning under all the innovations flooding ML
View GitHub Profile
@morgangiraud
morgangiraud / tinygrad-p2p-patched-driver.sh
Created May 2, 2024 11:56
tinygrad-p2p-patched-driver.sh
# this GIST is a follow-up to this previous GIST: https://gist.github.com/morgangiraud/ffa45e76b6891cd4e37e90d75b8be37b
# See the article here: https://morgangiraud.medium.com/multi-gpu-nvidia-p2p-capabilities-and-debugging-tips-fb7597b4e2b5
# It provides some tips and tricks to install Tinygrad patched nvidia open kernel to give P2P capabilities
# to the 40** series!
### Transitioning into complex operations, our aim is to minimize potential issues.
### Important: Verify that the version from nvidia-smi matches exactly what we intend to use.
### At the time of this writing, the reported version is 550.78.