Exemplos de get_cuda_version em Python

Exemplo n.º 1

Arquivo: transform.py Projeto: MultiTrickFox/Pytorch-0.4-_custom

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

Exemplo n.º 2

Arquivo: transform.py Projeto: gtgalone/pytorch

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