diff --git a/Dockerfile.cuda b/Dockerfile.cuda index 13fef89..4deb724 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -11,7 +11,7 @@ COPY . . ARG CUDACXX=/usr/local/cuda/bin/nvcc RUN cmake . -B ./build -DSD_CUDA=ON -RUN cmake --build ./build --config Release --parallel +RUN cmake --build ./build --config Release -j$(nproc) FROM nvidia/cuda:${CUDA_VERSION}-cudnn-runtime-ubuntu${UBUNTU_VERSION} AS runtime