pytorch #1366
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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', '3.12'] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: setup-python${{ matrix.python-version }} | |
| uses: actions/setup-python@v5 | |
| 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 |