Jackoatmon commited on
Commit
8d93a69
·
verified ·
1 Parent(s): 5674e6e

Update Feather H200 training runtime image

Browse files
Files changed (1) hide show
  1. 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 -m py_compile hydra/training.py prepare.py train.py && \
111
- bash -n scripts/run_domain_expanded_pretrain.sh
 
 
 
 
 
 
 
 
 
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} && \