diff --git a/.github/workflows/fvdb.yml b/.github/workflows/fvdb.yml index 2428475f0b..45c8a8cf59 100644 --- a/.github/workflows/fvdb.yml +++ b/.github/workflows/fvdb.yml @@ -78,7 +78,9 @@ jobs: - name: Build fvdb run: | - cd fvdb + cd fvdb; + free -g; + free -g | awk '/^Mem:/{jobs=int($4); print "Free memory (GB): " jobs}'; DISTUTILS_DEBUG="1" TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6+PTX" MAX_JOBS=$(($(nproc) < $(free -g | awk '/^Mem:/{jobs=int($4/2.5); if(jobs<1) jobs=1; print jobs}') ? $(nproc) : $(free -g | awk '/^Mem:/{jobs=int($4/2.5); if(jobs<1) jobs=1; print jobs}'))) python setup.py bdist_wheel --dist-dir=dist shell: bash -el {0}