diff --git a/pytorch/training/docker/2.8/py3/Dockerfile.cpu b/pytorch/training/docker/2.8/py3/Dockerfile.cpu index f3c32204a86b..9e305de1b54a 100644 --- a/pytorch/training/docker/2.8/py3/Dockerfile.cpu +++ b/pytorch/training/docker/2.8/py3/Dockerfile.cpu @@ -206,7 +206,7 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \ accelerate \ # pin numpy requirement for fastai dependency # requires explicit declaration of spacy, thic, blis - spacy \ + spacy==3.8.7 \ # pin thinc due to incompatibility with numpy 1.26.4 (sagemaker doesn't support latest numpy) thinc==8.3.4 \ blis \ diff --git a/pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu b/pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu index 52997f799aca..0b90890db2fb 100644 --- a/pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu +++ b/pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu @@ -126,7 +126,7 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \ accelerate \ # pin numpy requirement for fastai dependency # requires explicit declaration of spacy, thic, blis - spacy \ + spacy==3.8.7 \ # pin thinc due to incompatibility with numpy 1.26.4 (sagemaker doesn't support latest numpy) thinc==8.3.4 \ blis \