예제 #1
0
파일: misc.py 프로젝트: hyglvy/cstd 
def set_mode_gpu():
    """Set to the GPU mode. [**PyCaffe Style**]

    Returns
    -------
    None

    References
    ----------
    The implementation of `set_mode_gpu(_caffe.cpp, L52)`_.

    """
    config.EnableCUDA()
예제 #2
0
파일: common.py 프로젝트: zhangkaij/Dragon 
def set_mode_gpu():
    config.EnableCUDA()