docker_build_1.13.1_py3.9.16_cuda11.7.1_ubuntu20.04.yml

name: Build(1.13.1_py3.9.16_cuda11.7.1_ubuntu20.04)

env:
  BASE_IMAGE: "ubuntu:20.04"

  PYTHON_VERSION: "3.9.16"

  PYTORCH_VERSION: "1.13.1"
  PYTORCH_VERSION_SUFFIX: "+cu117"
  TORCHVISION_VERSION: "0.14.1"
  TORCHVISION_VERSION_SUFFIX: "+cu117"
  TORCHAUDIO_VERSION: "0.13.1"
  TORCHAUDIO_VERSION_SUFFIX: "+cu117"
  PYTORCH_DOWNLOAD_URL: "https://download.pytorch.org/whl/cu117/torch_stable.html"

  IMAGE_TAG: "1.13.1-py3.9.16-cuda11.7.1-ubuntu20.04"

on:
  push:
    branches:
      - main
    paths:
      - 'docker/ubuntu/**'
      - '.github/workflows/docker_build_1.13.1_py3.9.16_cuda11.7.1_ubuntu20.04.yml'

jobs:
  build:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v2

      - name: Login DockerHub
        run: docker login --username=${{ secrets.DOCKER_USERNAME }} --password=${{ secrets.DOCKER_PASSWORD }}

      - name: Build docker image
        run: docker/ubuntu/build.sh

      - name: Push docker image
        run: docker push cnstark/pytorch:${IMAGE_TAG}