Add swap for pytorch github workflow

bytedeco · May 5, 2024 · 8d13ded · 8d13ded
1 parent db813d6
commit 8d13ded
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/actions/deploy-ubuntu/action.yml b/.github/actions/deploy-ubuntu/action.yml
@@ -238,6 +238,13 @@ runs:
           $SUDO swapon /swapfile
         fi
 
+        if [[ "$CI_DEPLOY_MODULE" == "pytorch" ]]; then
+          $SUDO fallocate -l 4GB /swapfile
+          $SUDO chmod 600 /swapfile
+          $SUDO mkswap /swapfile
+          $SUDO swapon /swapfile
+        fi
+
         if [[ "$CI_DEPLOY_PLATFORM" == "linux-armhf" ]] && [[ "$CI_DEPLOY_MODULE" == "flycapture" ]]; then
           echo Installing FlyCapture amdhf
           python3 -m gdown.cli https://drive.google.com/uc?id=16NuUBs2MXQpVYqzDCEr9KdMng-6rHuDI