Skip to content

Commit c70fb8f

Browse files
authored
Patch PT 2.4 Training EC2 with new torchaudio (#4344)
* Patch PT 2.4 Training EC2 with new torchaudio * build test all * pin numpy * rebuild with no allowlist * test sm
1 parent 4d19f48 commit c70fb8f

6 files changed

+10
-9767
lines changed

pytorch/training/docker/2.4/py3/Dockerfile.cpu

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,10 @@ RUN pip install --no-cache-dir -U \
222222
fastai \
223223
accelerate \
224224
# pin numpy requirement for fastai dependency
225+
# requires explicit declaration of spacy, thic, blis
226+
spacy \
227+
thinc \
228+
blis \
225229
"numpy<2" \
226230
&& pip uninstall -y dataclasses
227231

@@ -282,6 +286,7 @@ RUN pip install --no-cache-dir -U \
282286
fastai \
283287
accelerate \
284288
# pin numpy requirement for fastai dependency
289+
# requires explicit declaration of spacy, thic, blis
285290
spacy \
286291
thinc \
287292
blis \

0 commit comments

Comments
 (0)