File size: 288 Bytes
5fa1a76 |
1 2 3 4 5 6 7 8 |
pip install pytest-instafail pytest --instafail To GPU or not to GPU On a GPU-enabled setup, to test in CPU-only mode add CUDA_VISIBLE_DEVICES="": CUDA_VISIBLE_DEVICES="" pytest tests/utils/test_logging.py or if you have multiple gpus, you can specify which one is to be used by pytest. |