Esempi in Python per get_gpu_names

Esempio n. 1

File: SprintInterface.py Progetto: 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")

Esempio n. 2

File: SprintInterface.py Progetto: 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")

Esempio n. 3

File: SprintInterface.py Progetto: 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)