Skip to content

pytorch

pytorch #1067

Workflow file for this run

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