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")