Exemplos de get_gpu_info em Python

Exemplo n.º 1

Arquivo: astra_recon.py Projeto: Pasarus/mantidimaging

 def _count_gpus() -> int:
     num_gpus = 0
     msg = ''
     while "Invalid device" not in msg:
         num_gpus += 1
         msg = astra.get_gpu_info(num_gpus)
     return num_gpus

Exemplo n.º 2

Arquivo: tests.py Projeto: astra-toolbox/astra-toolbox

def test_CUDA():
  """Perform a very basic functionality test, including CUDA"""
  
  import astra
  print("Getting GPU info... ", end="")
  print(astra.get_gpu_info())
  ok1 = _basic_par2d()
  ok2 = _basic_par2d_cuda()
  ok3 = _basic_par3d_cuda()
  if not (ok1 and ok2 and ok3):
    raise RuntimeError("Test failed")

Exemplo n.º 3

def test_CUDA():
    """Perform a very basic functionality test, including CUDA"""

    import astra
    print("Getting GPU info... ", end="")
    print(astra.get_gpu_info())
    ok1 = _basic_par2d()
    ok2 = _basic_par2d_cuda()
    ok3 = _basic_par3d_cuda()
    if not (ok1 and ok2 and ok3):
        raise RuntimeError("Test failed")