__init__.py

import sys
import core
__all__ = ['proto']
argv = []
if core.is_compile_gpu():
    argv = list(sys.argv) + [
        "--tryfromenv=fraction_of_gpu_memory_to_use,use_pinned_memory"
    ]
else:
    argv = list(sys.argv) + ["--tryfromenv=use_pinned_memory"]
core.init_gflags(argv)