Skip to content

Commit b54ed63

Browse files
authored
Fix PyTorch 2.8 images (#5490)
* pin spacy version to fix package regression in PyTorch 2.8 images
1 parent bc35308 commit b54ed63

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \
206206
accelerate \
207207
# pin numpy requirement for fastai dependency
208208
# requires explicit declaration of spacy, thic, blis
209-
spacy \
209+
spacy==3.8.7 \
210210
# pin thinc due to incompatibility with numpy 1.26.4 (sagemaker doesn't support latest numpy)
211211
thinc==8.3.4 \
212212
blis \

pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \
126126
accelerate \
127127
# pin numpy requirement for fastai dependency
128128
# requires explicit declaration of spacy, thic, blis
129-
spacy \
129+
spacy==3.8.7 \
130130
# pin thinc due to incompatibility with numpy 1.26.4 (sagemaker doesn't support latest numpy)
131131
thinc==8.3.4 \
132132
blis \

0 commit comments

Comments
 (0)