tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

push test build

Browse Source
This commit is contained in:
Wing Lian
2024-12-06 20:15:02 -05:00
parent ec819dde3b
commit e64c32c0bd
1 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
.github/workflows/base.yml vendored
View File

@@ -22,30 +22,30 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
include: include:
- cuda: "121" # - cuda: "121"
cuda_version: 12.1.1 # cuda_version: 12.1.1
cudnn_version: 8 # cudnn_version: 8
python_version: "3.10" # python_version: "3.10"
pytorch: 2.3.1 # pytorch: 2.3.1
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" # torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
from_base_img: "" # from_base_img: ""
from_base_tag: "" # from_base_tag: ""
- cuda: "121" # - cuda: "121"
cuda_version: 12.1.1 # cuda_version: 12.1.1
cudnn_version: 8 # cudnn_version: 8
python_version: "3.11" # python_version: "3.11"
pytorch: 2.3.1 # pytorch: 2.3.1
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" # torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
from_base_img: "" # from_base_img: ""
from_base_tag: "" # from_base_tag: ""
- cuda: "124" # - cuda: "124"
cuda_version: 12.4.1 # cuda_version: 12.4.1
cudnn_version: "" # cudnn_version: ""
python_version: "3.11" # python_version: "3.11"
pytorch: 2.4.1 # pytorch: 2.4.1
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" # torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
from_base_img: "" # from_base_img: ""
from_base_tag: "" # from_base_tag: ""
- cuda: "124" - cuda: "124"
cuda_version: 12.4.1 cuda_version: 12.4.1
cudnn_version: "" cudnn_version: ""
@@ -76,6 +76,7 @@ jobs:
with: with:
context: . context: .
file: ./docker/Dockerfile-base file: ./docker/Dockerfile-base
push: true
# push: ${{ github.event_name != 'pull_request' }} # 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 }} 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 }} labels: ${{ steps.metadata.outputs.labels }}