Python init示例

示例#1

文件： pmat.py 项目： Hanyi-Li/urbansim

def initialize_gpu():
    from cudamat import cudamat as cm
    global cm

    # this class wraps matrix operations so that you can switch between numpy and cuda operations
    cm.cuda_set_device(0)
    cm.init()

示例#2

文件： pmat.py 项目： mdroid/urbansim

def initialize_gpu():
    from cudamat import cudamat as cm
    global cm

    # this class wraps matrix operations so that you can switch between numpy and cuda operations
    cm.cuda_set_device(0)
    cm.init()