Python global_torch_device Examples

Example #1

File: cache_sessions.py Project: cnheider/draugr

 def a():
     """ """
     print(torch.cuda.memory_cached(global_torch_device()))
     with TorchCacheSession():
         torch.tensor([0.0], device=global_torch_device())
         print(torch.cuda.memory_cached(global_torch_device()))
     print(torch.cuda.memory_cached(global_torch_device()))

Example #2

File: cache_sessions.py Project: cnheider/draugr

 def e():
     """ """
     print(
         global_torch_device(override=global_torch_device(
             device_preference=True)))
     print(global_torch_device())
     with TorchCpuSession():
         print(global_torch_device())
     print(global_torch_device())

Example #3

File: cache_sessions.py Project: cnheider/draugr

 def d():
     """ """
     print(
         global_torch_device(override=global_torch_device(
             device_preference=False)))
     print(global_torch_device())
     with TorchCudaSession():
         print(global_torch_device())
     print(global_torch_device())

Example #4

File: cache_sessions.py Project: pything/draugr

 def __init__(self,
              using_cuda: bool = global_torch_device().type == "cuda"):
     self.using_cuda = using_cuda