mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-11 12:31:23 +08:00
Adrian WälchliandJirka Borovec
a6de1b8d75
* add doctest to circleci * Revert "add doctest to circleci" This reverts commit c45b34ea911a81f87989f6c3a832b1e8d8c471c6. * Revert "Revert "add doctest to circleci"" This reverts commit 41fca97fdcfe1cf4f6bdb3bbba75d25fa3b11f70. * doctest docs rst files * Revert "doctest docs rst files" This reverts commit b4a2e83e3da5ed1909de500ec14b6b614527c07f. * doctest only rst * doctest debugging.rst * doctest apex * doctest callbacks * doctest early stopping * doctest for child modules * doctest experiment reporting * indentation * doctest fast training * doctest for hyperparams * doctests for lr_finder * doctests multi-gpu * more doctest * make doctest drone * fix label build error * update fast training * update invalid imports * fix problem with int device count * rebase stuff * wip * wip * wip * intro guide * add missing code block * circleci * logger import for doctest * test if doctest runs on drone * fix mnist download * also run install deps for building docs * install cmake * try sudo * hide output * try pip stuff * try to mock horovod * Tranfer -> Transfer * add torchvision to extras * revert pip stuff * mlflow file location * do not mock torch * torchvision * drone extra req. * try higher sphinx version * Revert "try higher sphinx version" This reverts commit 490ac28e46d6fd52352640dfdf0d765befa56988. * try coverage command * try coverage command * try undoc flag * newline * undo drone * report coverage * review Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * remove torchvision from extras * skip tests only if torchvision not available * fix testoutput torchvision Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
59 lines
1.7 KiB
YAML
59 lines
1.7 KiB
YAML
# https://docs.drone.io/pipeline/docker/examples/languages/python/#python-example
|
|
|
|
kind: pipeline
|
|
type: docker
|
|
name: torch-GPU
|
|
|
|
steps:
|
|
- name: testing
|
|
image: pytorchlightning/pytorch_lightning:devel-pt_1_4
|
|
|
|
environment:
|
|
SLURM_LOCALID: 0
|
|
CODECOV_TOKEN:
|
|
from_secret: codecov_token
|
|
HOROVOD_GPU_ALLREDUCE: NCCL
|
|
HOROVOD_GPU_BROADCAST: NCCL
|
|
HOROVOD_WITH_PYTORCH: 1
|
|
HOROVOD_WITHOUT_TENSORFLOW: 1
|
|
HOROVOD_WITHOUT_MXNET: 1
|
|
HOROVOD_WITH_GLOO: 1
|
|
HOROVOD_WITHOUT_MPI: 1
|
|
|
|
#volumes:
|
|
# # Mount pip cache from host
|
|
# - name: pip_cache
|
|
# path: /opt/conda/lib/python3.7/site-packages
|
|
|
|
commands:
|
|
- export PATH="$PATH:/root/.local/bin"
|
|
- python --version
|
|
- pip install pip -U
|
|
- pip --version
|
|
- nvidia-smi
|
|
#- bash ./tests/install_AMP.sh
|
|
- apt-get update && apt-get install -y cmake
|
|
- pip install -r requirements.txt --user -q
|
|
- pip install -r ./tests/requirements-devel.txt --user -q
|
|
#- pip install -r ./docs/requirements.txt --user -q
|
|
- pip list
|
|
- python -c "import torch ; print(' & '.join([torch.cuda.get_device_name(i) for i in range(torch.cuda.device_count())]) if torch.cuda.is_available() else 'only CPU')"
|
|
- coverage run --source pytorch_lightning -m py.test pytorch_lightning tests benchmarks -v --doctest-modules # --flake8
|
|
#- cd docs; make doctest; make coverage
|
|
- coverage report
|
|
- codecov --token $CODECOV_TOKEN # --pr $DRONE_PULL_REQUEST --build $DRONE_BUILD_NUMBER --branch $DRONE_BRANCH --commit $DRONE_COMMIT --tag $DRONE_TAG
|
|
- python tests/collect_env_details.py
|
|
|
|
trigger:
|
|
branch:
|
|
- master
|
|
event:
|
|
include:
|
|
- push
|
|
- pull_request
|
|
|
|
#volumes:
|
|
# - name: pip_cache
|
|
# host:
|
|
# path: /tmp/cache/drone/pip
|