Esempi in Python per get_cuda_version

Esempio n. 1

File: transform.py Progetto: MultiTrickFox/Pytorch-0.4-_custom

def check_gpu_():
    try:
        C.get_cuda_version()
    except Exception as _:
        raise Exception("TensorRT related functions require CUDA support")

Esempio n. 2

File: transform.py Progetto: gtgalone/pytorch

def check_gpu_():
    try:
        C.get_cuda_version()
    except Exception as _:
       raise Exception("TensorRT related functions require CUDA support")