chainer.testing.attr.multi_gpu¶
- chainer.testing.attr.multi_gpu(gpu_num)[source]¶
Decorator to indicate number of GPUs required to run the test.
Tests can be annotated with this decorator (e.g.,
@multi_gpu(2)
) to declare number of GPUs required to run. When running tests, ifCHAINER_TEST_GPU_LIMIT
environment variable is set to value greater than or equals to 0, test cases that require GPUs more than the limit will be skipped.