diff --git a/Dockerfile b/Dockerfile index e98867f..a549264 100644 --- a/Dockerfile +++ b/Dockerfile @@ -35,7 +35,7 @@ ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF- RUN apt-get update && apt-get install -y --no-install-recommends \ python3 python3-pip python3-distutils git build-essential \ cmake ninja-build libjpeg-dev libpng-dev ca-certificates \ - libopenmpi-dev \ + libopenmpi-dev libopenblas-dev\ libnccl2=2.22.3-1+cuda12.6 \ libnccl-dev=2.22.3-1+cuda12.6 && \ python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools