nv-lightning-v100 #204
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: nv-lightning-v100 | |
| on: | |
| push: | |
| branches: | |
| - 'staging**' | |
| paths-ignore: | |
| - 'docs/**' | |
| - 'blogs/**' | |
| pull_request: | |
| paths-ignore: | |
| - 'docs/**' | |
| - 'blogs/**' | |
| schedule: | |
| - cron: "0 0 * * *" | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| unit-tests: | |
| runs-on: [self-hosted, nvidia, cu111, v100] | |
| steps: | |
| - uses: actions/checkout@v3 | |
| - id: setup-venv | |
| uses: ./.github/workflows/setup-venv | |
| - name: Install pytorch | |
| run: | | |
| pip install --no-cache-dir torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html | |
| python -c "import torch; print('torch:', torch.__version__, torch)" | |
| python -c "import torch; print('CUDA available:', torch.cuda.is_available())" | |
| - name: Install deepspeed | |
| run: | | |
| pip install .[dev,autotuning] | |
| ds_report | |
| - name: Python environment | |
| run: | | |
| pip list | |
| - name: PyTorch Lightning Tests | |
| run: | | |
| if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi | |
| # Pin pytorch-lightning version to latest pre-2.0.0+ as these require updating the pinned torch versions above. | |
| pip install pytorch-lightning==1.9.4 | |
| pip install "protobuf<4.21.0" | |
| cd tests | |
| TORCH_EXTENSIONS_DIR=./torch-extensions pytest --color=yes --durations=0 --verbose lightning/ |