Spaces:
Build error
Build error
Update Feather a10g-large-gt80k training runtime image
Browse files- Dockerfile +0 -2
- entrypoint.py +9 -0
- 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
|