tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-05-02 14:00:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(install): add nvidia-nvshmem to GPU extras to prevent libnvshmem_host.so.3 error

Browse Source
This commit is contained in:
Vishal Krishna Kumar 2025-12-28 23:57:16 -08:00
parent 8f979a8bda
commit 8b8e94c632
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@ -66,6 +66,8 @@ cpu = [
]
gpu = [
"torch>=2.8.0",
"nvidia-nvshmem-cu12>=3.3",
]
[tool.uv]