icarus112 commited on
Commit
f14e3d5
·
verified ·
1 Parent(s): 45ed9ea

Update Feather a10g-large-gt80k training runtime image

Browse files
Files changed (3) hide show
  1. Dockerfile +0 -2
  2. entrypoint.py +9 -0
  3. patch.zip +3 -0
Dockerfile CHANGED
@@ -118,7 +118,5 @@ RUN python -m py_compile hydra/training.py prepare.py train.py && \
118
 
119
  RUN export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH} && \
120
  echo "building htm_rust GPU kernels for HTM_CUDA_ARCH=${HTM_CUDA_ARCH} TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}" && \
121
- # maturin build --release --features gpu --manifest-path htm_rust/Cargo.toml && \
122
- # pip install htm_rust/target/wheels/htm_rust-*.whl
123
 
124
  CMD ["python", "/app/entrypoint.py"]
 
118
 
119
  RUN export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH} && \
120
  echo "building htm_rust GPU kernels for HTM_CUDA_ARCH=${HTM_CUDA_ARCH} TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}" && \
 
 
121
 
122
  CMD ["python", "/app/entrypoint.py"]
entrypoint.py CHANGED
@@ -203,6 +203,15 @@ def _wait_for_cuda_ready(timeout_s: int = 120) -> None:
203
  def run_job_mode() -> int:
204
  os.chdir(REPO_ROOT)
205
 
 
 
 
 
 
 
 
 
 
206
  # Dynamic live patch from GitHub to bypass Space build errors
207
  GIT_REF = os.environ.get('FEATHER_GIT_REF')
208
  if GIT_REF:
 
203
  def run_job_mode() -> int:
204
  os.chdir(REPO_ROOT)
205
 
206
+ # Unpack python-only patches from overlay
207
+ PATCH_ZIP = Path('/workspace/feather/patch.zip')
208
+ if PATCH_ZIP.exists():
209
+ print(f'[bootstrap] applying hotpatch from {PATCH_ZIP}...', flush=True)
210
+ import zipfile
211
+ with zipfile.ZipFile(PATCH_ZIP, 'r') as zip_ref:
212
+ zip_ref.extractall('/workspace/feather')
213
+
214
+
215
  # Dynamic live patch from GitHub to bypass Space build errors
216
  GIT_REF = os.environ.get('FEATHER_GIT_REF')
217
  if GIT_REF:
patch.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1148711624bd972a937d2cc9835d1406af2ccdb224951de59f07d80b1d81fc88
3
+ size 165518