Skip to content

build(deps): bump transformers from 4.53.0 to 5.5.0 in /test/images/nvidia-inference #10

build(deps): bump transformers from 4.53.0 to 5.5.0 in /test/images/nvidia-inference

build(deps): bump transformers from 4.53.0 to 5.5.0 in /test/images/nvidia-inference #10

name: "Nvidia Images CI"
on:
pull_request:
types:
- opened
- reopened
- synchronize
paths:
- 'test/images/nvidia**'
jobs:
build-image-nvidia:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- run: ./hack/free-disk-space.sh
- run: docker build --file test/images/nvidia/Dockerfile .
build-image-nvidia-training:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- run: ./hack/free-disk-space.sh
- run: |
docker build --file test/images/nvidia-training/Dockerfile test/images/nvidia-training \
--build-arg PYTORCH_BUILD_ENV="MAX_JOBS=$(($(nproc) - 2)) USE_MKLDNN=0 USE_DISTRIBUTED=0 USE_CUDA=0 USE_ROCM=0 USE_CAFFE2=0 USE_QNNPACK=0 USE_NNPACK=0 USE_XNNPACK=0 USE_MPS=0 BUILD_SHARED_LIBS=OFF USE_FLASH_ATTENTION=0 USE_MEM_EFF_ATTENTION=0 BUILD_TEST=0"
build-image-nvidia-inference:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- run: ./hack/free-disk-space.sh
- run: |
docker build --file test/images/nvidia-inference/Dockerfile test/images/nvidia-inference \
--build-arg PYTORCH_BUILD_ENV="MAX_JOBS=$(($(nproc) - 2)) USE_MKLDNN=0 USE_DISTRIBUTED=0 USE_CUDA=0 USE_ROCM=0 USE_CAFFE2=0 USE_QNNPACK=0 USE_NNPACK=0 USE_XNNPACK=0 USE_MPS=0 BUILD_SHARED_LIBS=OFF USE_FLASH_ATTENTION=0 USE_MEM_EFF_ATTENTION=0 BUILD_TEST=0"