Python get_cuda_version示例

示例#1

文件： transform.py 项目： MultiTrickFox/Pytorch-0.4-_custom

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

示例#2

文件： transform.py 项目： gtgalone/pytorch

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