diff --git a/.github/workflows/build-docker-images.yml b/.github/workflows/build-docker-images.yml index 281e374f..7ffef96d 100644 --- a/.github/workflows/build-docker-images.yml +++ b/.github/workflows/build-docker-images.yml @@ -14,21 +14,9 @@ env: jobs: latest-cpu: name: CPU - runs-on: ubuntu-latest + runs-on: + group: aws-general-8-plus steps: - - name: Cleanup disk - run: | - sudo df -h - # sudo ls -l /usr/local/lib/ - # sudo ls -l /usr/share/ - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo rm -rf /usr/local/lib/android - sudo rm -rf /usr/share/dotnet - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo df -h - - name: Set up Docker Buildx uses: docker/setup-buildx-action@v3 @@ -55,20 +43,9 @@ jobs: latest-cuda: name: GPU - runs-on: ubuntu-latest + runs-on: + group: aws-general-8-plus steps: - - name: Cleanup disk - run: | - sudo df -h - # sudo ls -l /usr/local/lib/ - # sudo ls -l /usr/share/ - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo rm -rf /usr/local/lib/android - sudo rm -rf /usr/share/dotnet - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo df -h - name: Set up Docker Buildx uses: docker/setup-buildx-action@v3 @@ -95,20 +72,9 @@ jobs: latest-cuda-dev: name: GPU Dev - runs-on: ubuntu-latest + runs-on: + group: aws-general-8-plus steps: - - name: Cleanup disk - run: | - sudo df -h - # sudo ls -l /usr/local/lib/ - # sudo ls -l /usr/share/ - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo rm -rf /usr/local/lib/android - sudo rm -rf /usr/share/dotnet - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo df -h - name: Set up Docker Buildx uses: docker/setup-buildx-action@v3 diff --git a/.github/workflows/nightly-tests.yml b/.github/workflows/nightly-tests.yml index b3a2157b..f967533a 100644 --- a/.github/workflows/nightly-tests.yml +++ b/.github/workflows/nightly-tests.yml @@ -16,7 +16,8 @@ jobs: name: CPU strategy: fail-fast: false - runs-on: ubuntu-latest + runs-on: + group: aws-general-8-plus container: image: huggingface/lerobot-cpu:latest options: --shm-size "16gb" @@ -43,7 +44,8 @@ jobs: name: GPU strategy: fail-fast: false - runs-on: [single-gpu, nvidia-gpu, t4, ci] + runs-on: + group: aws-g6-4xlarge-plus env: CUDA_VISIBLE_DEVICES: "0" TEST_TYPE: "single_gpu" diff --git a/.github/workflows/test-docker-build.yml b/.github/workflows/test-docker-build.yml index 24c2b2a0..979897b0 100644 --- a/.github/workflows/test-docker-build.yml +++ b/.github/workflows/test-docker-build.yml @@ -42,26 +42,14 @@ jobs: build_modified_dockerfiles: name: Build modified Docker images needs: get_changed_files - runs-on: ubuntu-latest + runs-on: + group: aws-general-8-plus if: ${{ needs.get_changed_files.outputs.matrix }} != '' strategy: fail-fast: false matrix: docker-file: ${{ fromJson(needs.get_changed_files.outputs.matrix) }} steps: - - name: Cleanup disk - run: | - sudo df -h - # sudo ls -l /usr/local/lib/ - # sudo ls -l /usr/share/ - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo rm -rf /usr/local/lib/android - sudo rm -rf /usr/share/dotnet - sudo du -sh /usr/local/lib/ - sudo du -sh /usr/share/ - sudo df -h - - name: Set up Docker Buildx uses: docker/setup-buildx-action@v3