diff --git a/docker/pyt_huggingface.ubuntu.amd.Dockerfile b/docker/pyt_huggingface.ubuntu.amd.Dockerfile index 3ab9eaa..2b7be15 100644 --- a/docker/pyt_huggingface.ubuntu.amd.Dockerfile +++ b/docker/pyt_huggingface.ubuntu.amd.Dockerfile @@ -66,4 +66,5 @@ RUN locale-gen en_US.UTF-8 # record configuration for posterity RUN pip3 list - +# install numpy < 2.0 to avoid compatibility issues +RUN pip3 install "numpy<2"