diff --git a/dockers/base-conda/Dockerfile b/dockers/base-conda/Dockerfile index 6ac4b2c0e6bee..c596f5c6a8c5b 100644 --- a/dockers/base-conda/Dockerfile +++ b/dockers/base-conda/Dockerfile @@ -68,8 +68,8 @@ ENV \ HOROVOD_WITHOUT_MXNET=1 \ HOROVOD_WITH_GLOO=1 \ HOROVOD_WITHOUT_MPI=1 \ - # MAKEFLAGS="-j$(nproc)" \ - MAKEFLAGS="-j1" \ + MAKEFLAGS="-j$(nproc)" \ + # MAKEFLAGS="-j1" \ TORCH_CUDA_ARCH_LIST="3.7;5.0;6.0;7.0;7.5" \ CONDA_ENV=lightning diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile index cf602be72352b..e16971bdc2a1a 100644 --- a/dockers/base-cuda/Dockerfile +++ b/dockers/base-cuda/Dockerfile @@ -74,8 +74,8 @@ ENV \ HOROVOD_WITHOUT_MXNET=1 \ HOROVOD_WITH_GLOO=1 \ HOROVOD_WITHOUT_MPI=1 \ - # MAKEFLAGS="-j$(nproc)" \ - MAKEFLAGS="-j1" \ + MAKEFLAGS="-j$(nproc)" \ + # MAKEFLAGS="-j1" \ TORCH_CUDA_ARCH_LIST="3.7;5.0;6.0;7.0;7.5" COPY ./requirements.txt requirements.txt