Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
41f4e02
1
Parent(s):
734af3d
docs: Add system dependencies to generate_summaries_uv.py
Browse filesAdded documentation for required system dependencies (python3-dev and gcc)
that are needed for Triton compilation when running in Docker containers.
Updated the example hfjobs command to install these dependencies.
- generate_summaries_uv.py +5 -0
generate_summaries_uv.py
CHANGED
|
@@ -12,6 +12,10 @@
|
|
| 12 |
# "tqdm",
|
| 13 |
# ]
|
| 14 |
# ///
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
import argparse
|
| 17 |
import logging
|
|
@@ -231,6 +235,7 @@ if __name__ == "__main__":
|
|
| 231 |
print(
|
| 232 |
"hfjobs run --flavor l4x1 --secret HF_TOKEN=hf_*** ghcr.io/astral-sh/uv:debian /bin/bash -c '"
|
| 233 |
)
|
|
|
|
| 234 |
print("export HOME=/tmp && \\")
|
| 235 |
print("export USER=dummy && \\")
|
| 236 |
print("export TORCHINDUCTOR_CACHE_DIR=/tmp/torch-inductor && \\")
|
|
|
|
| 12 |
# "tqdm",
|
| 13 |
# ]
|
| 14 |
# ///
|
| 15 |
+
#
|
| 16 |
+
# System dependencies required:
|
| 17 |
+
# - python3-dev (Python.h headers for Triton compilation)
|
| 18 |
+
# - gcc (C compiler for Triton)
|
| 19 |
|
| 20 |
import argparse
|
| 21 |
import logging
|
|
|
|
| 235 |
print(
|
| 236 |
"hfjobs run --flavor l4x1 --secret HF_TOKEN=hf_*** ghcr.io/astral-sh/uv:debian /bin/bash -c '"
|
| 237 |
)
|
| 238 |
+
print("apt-get update && apt-get install -y python3-dev gcc && \\")
|
| 239 |
print("export HOME=/tmp && \\")
|
| 240 |
print("export USER=dummy && \\")
|
| 241 |
print("export TORCHINDUCTOR_CACHE_DIR=/tmp/torch-inductor && \\")
|