Python get_gpu_names Examples

Example #1

File: SprintInterface.py Project: tazdriver/returnn

def dumpFlags():
    print "available GPUs:", get_gpu_names()

    import theano.sandbox.cuda as theano_cuda
    print "CUDA via", theano_cuda.__file__
    print "CUDA available:", theano_cuda.cuda_available

    print "THEANO_FLAGS:", rnn.TheanoFlags
    print "CUDA_LAUNCH_BLOCKING:", os.environ.get("CUDA_LAUNCH_BLOCKING")

Example #2

File: SprintInterface.py Project: chagge/returnn

def dumpFlags():
  print "available GPUs:", get_gpu_names()

  import theano.sandbox.cuda as theano_cuda
  print "CUDA via", theano_cuda.__file__
  print "CUDA available:", theano_cuda.cuda_available

  print "THEANO_FLAGS:", rnn.TheanoFlags
  print "CUDA_LAUNCH_BLOCKING:", os.environ.get("CUDA_LAUNCH_BLOCKING")

Example #3

File: SprintInterface.py Project: wbengine/returnn

def dumpFlags():
    print("CUDA_VISIBLE_DEVICES:", os.environ.get("CUDA_VISIBLE_DEVICES"))
    print("CUDA_LAUNCH_BLOCKING:", os.environ.get("CUDA_LAUNCH_BLOCKING"))

    if BackendEngine.is_theano_selected():
        print("available GPUs:", get_gpu_names())

        from theano.sandbox import cuda as theano_cuda
        print("CUDA via", theano_cuda.__file__)
        print("CUDA available:", theano_cuda.cuda_available)

        print("THEANO_FLAGS:", rnn.TheanoFlags)