davanstrien HF Staff commited on
Commit
41f4e02
·
1 Parent(s): 734af3d

docs: Add system dependencies to generate_summaries_uv.py

Browse files

Added 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.

Files changed (1) hide show
  1. 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 && \\")