Python get_gpu_names Exemples

Exemple #1

Fichier : SprintInterface.py Projet : 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")

Exemple #2

Fichier : SprintInterface.py Projet : 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")

Exemple #3

Fichier : SprintInterface.py Projet : 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)