Spaces:
Runtime error
Runtime error
Update Feather H200 training runtime image
Browse files- Dockerfile +11 -2
Dockerfile
CHANGED
|
@@ -107,8 +107,17 @@ COPY overlay /workspace/feather
|
|
| 107 |
COPY entrypoint.py /app/entrypoint.py
|
| 108 |
WORKDIR /workspace/feather
|
| 109 |
|
| 110 |
-
RUN python -
|
| 111 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 112 |
|
| 113 |
RUN export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH} && \
|
| 114 |
export HTM_CUDA_ARCH=${HTM_CUDA_ARCH} && \
|
|
|
|
| 107 |
COPY entrypoint.py /app/entrypoint.py
|
| 108 |
WORKDIR /workspace/feather
|
| 109 |
|
| 110 |
+
RUN python - <<'PY'
|
| 111 |
+
from pathlib import Path
|
| 112 |
+
for sh in Path('/workspace/feather/scripts').glob('*.sh'):
|
| 113 |
+
raw = sh.read_bytes()
|
| 114 |
+
norm = raw.replace(b'\r\n', b'\n')
|
| 115 |
+
if norm != raw:
|
| 116 |
+
sh.write_bytes(norm)
|
| 117 |
+
PY
|
| 118 |
+
|
| 119 |
+
RUN python -m py_compile hydra/training.py prepare.py train.py && \
|
| 120 |
+
bash -n scripts/run_domain_expanded_pretrain.sh
|
| 121 |
|
| 122 |
RUN export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH} && \
|
| 123 |
export HTM_CUDA_ARCH=${HTM_CUDA_ARCH} && \
|