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 && \\")
|