diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 7c2488b64..6452b2fad 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -22,30 +22,30 @@ jobs: fail-fast: false matrix: include: - - cuda: "121" - cuda_version: 12.1.1 - cudnn_version: 8 - python_version: "3.10" - pytorch: 2.3.1 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - from_base_img: "" - from_base_tag: "" - - cuda: "121" - cuda_version: 12.1.1 - cudnn_version: 8 - python_version: "3.11" - pytorch: 2.3.1 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - from_base_img: "" - from_base_tag: "" - - cuda: "124" - cuda_version: 12.4.1 - cudnn_version: "" - python_version: "3.11" - pytorch: 2.4.1 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - from_base_img: "" - from_base_tag: "" +# - cuda: "121" +# cuda_version: 12.1.1 +# cudnn_version: 8 +# python_version: "3.10" +# pytorch: 2.3.1 +# torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" +# from_base_img: "" +# from_base_tag: "" +# - cuda: "121" +# cuda_version: 12.1.1 +# cudnn_version: 8 +# python_version: "3.11" +# pytorch: 2.3.1 +# torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" +# from_base_img: "" +# from_base_tag: "" +# - cuda: "124" +# cuda_version: 12.4.1 +# cudnn_version: "" +# python_version: "3.11" +# pytorch: 2.4.1 +# torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" +# from_base_img: "" +# from_base_tag: "" - cuda: "124" cuda_version: 12.4.1 cudnn_version: "" @@ -76,6 +76,7 @@ jobs: with: context: . file: ./docker/Dockerfile-base + push: true # push: ${{ github.event_name != 'pull_request' }} tags: ${{ steps.metadata.outputs.tags }}-base-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }} labels: ${{ steps.metadata.outputs.labels }}