SwitchProducerCUDA Namespace Reference

Classes
class	SwitchProducerCUDA
class	TestSwitchProducerCUDA

Functions
def	_switch_cuda

Variables
	_cuda_enabled_cached = None

Function Documentation

def SwitchProducerCUDA._switch_cuda ( )

private

Definition at line 5 of file SwitchProducerCUDA.py.

def _switch_cuda():
    global _cuda_enabled_cached
    if _cuda_enabled_cached is None:
        import os
        _cuda_enabled_cached = (os.system("cudaIsEnabled") == 0)
    return (_cuda_enabled_cached, 2)