pytorch.yml

name: pytorch

on:
  schedule:
    - cron: '0 15 * * *'
  pull_request:
    paths:
      - 'pytorch/**'
      - '.github/workflows/pytorch.yml'

jobs:
  examples:
    if: (github.event_name == 'schedule' && github.repository == 'optuna/optuna-examples') || (github.event_name != 'schedule')
    runs-on: ubuntu-latest
    strategy:
      matrix:
        python-version: ['3.8', '3.9', '3.10', '3.11']

    steps:
    - uses: actions/checkout@v3
    - name: setup-python${{ matrix.python-version }}
      uses: actions/setup-python@v4
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install (apt)
      run: |
        sudo apt-get update
        sudo apt-get -y install openmpi-bin libopenmpi-dev
    - name: Install (Python)
      run: |
        python -m pip install --upgrade pip
        pip install --progress-bar off -U setuptools
        pip install git+https://github.com/optuna/optuna.git
        python -c 'import optuna'
        pip install git+https://github.com/optuna/optuna-integration.git
        python -c 'import optuna_integration'

        pip install -r pytorch/requirements.txt
    - name: Run PyTorch examples
      run: |
        python pytorch/pytorch_simple.py
      env:
        OMP_NUM_THREADS: 1
    - name: Run PyTorch checkpoint example
      run: |
        set +e
        timeout 20 python pytorch/pytorch_checkpoint.py > /dev/null
        set -e
        python pytorch/pytorch_checkpoint.py > /dev/null
      env:
        OMP_NUM_THREADS: 1
    - name: Run multi-node examples
      run: |
        export OMPI_MCA_rmaps_base_oversubscribe=yes
        STORAGE_URL=sqlite:///example.db
        mpirun -n 2 -- python pytorch/pytorch_distributed_simple.py
      env:
        OMP_NUM_THREADS: 1
    - name: Run PyTorch spawn examples
      run: |
        python pytorch/pytorch_distributed_spawn.py --no-cuda
      env:
        OMP_NUM_THREADS: 1
    - name: Run PyTorch Ignite examples
      run: |
        python pytorch/pytorch_ignite_simple.py
      env:
        OMP_NUM_THREADS: 1
    - name: Run PyTorch Lightning examples
      run: |
        python pytorch/pytorch_lightning_simple.py
      env:
        OMP_NUM_THREADS: 1
    - name: Run PyTorch Lightning DDP examples
      run: |
        python pytorch/pytorch_lightning_ddp.py
        python pytorch/pytorch_lightning_ddp.py -p
      env:
        OMP_NUM_THREADS: 1