diff --git a/Dockerfile b/Dockerfile index a549264..ebe0130 100644 --- a/Dockerfile +++ b/Dockerfile @@ -33,7 +33,7 @@ FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS builder-extras ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8 RUN apt-get update && apt-get install -y --no-install-recommends \ - python3 python3-pip python3-distutils git build-essential \ + python3 python3-pip python3-distutils python3.10-dev git build-essential \ cmake ninja-build libjpeg-dev libpng-dev ca-certificates \ libopenmpi-dev libopenblas-dev\ libnccl2=2.22.3-1+cuda12.6 \