Python obtain_lock_idの例

コード例 #1

ファイル: gpu.py プロジェクト: jakesnell/myshkin

def get_board():
    board_id = gg.obtain_lock_id()
    if board_id < 0:
        time.sleep(10)
        board_id = gg.obtain_lock_id() # try again
        if board_id < 0:
            raise Exception("could not obtain lock")
        else:
            return board_id
    else:
        return board_id

コード例 #2

def LockGPU(max_retries=10, board=-1):
    retry_count = 0
    while board == -1 and retry_count < max_retries:
        board = gpu_lock.obtain_lock_id()
        if board == -1:
            sleep(1)
            retry_count += 1
    if board == -1:
        print 'No GPU board available.'
        sys.exit(1)
    else:
        cm.cuda_set_device(board)
        cm.cublas_init()
    return board

コード例 #3

ファイル: util.py プロジェクト: Luna86/RNN-DDCRP

def LockGPU(max_retries=10, board=-1):
  retry_count = 0
  while board == -1 and retry_count < max_retries:
    board = gpu_lock.obtain_lock_id()
    if board == -1:
      sleep(1)
      retry_count += 1
  if board == -1:
    print 'No GPU board available.'
    sys.exit(1)
  else:
    cm.cuda_set_device(board)
    cm.cublas_init()
  return board