예제 #1
0
파일: 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
0
파일: 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()