Function find_cuda_path

python/tvm/support/nvcc.py:701–720

Utility function to find CUDA path Returns ------- path : str Path to CUDA root.

()

def find_cuda_path():
    """Utility function to find CUDA path

    Returns
    -------
    path : str
        Path to CUDA root.
    """
    if "CUDA_PATH" in os.environ:
        return os.environ["CUDA_PATH"]
    cmd = ["which", "nvcc"]
    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    (out, _) = proc.communicate()
    out = out.decode("utf-8", errors="replace")
    if proc.returncode == 0:
        return os.path.realpath(os.path.join(str(out).strip(), "../.."))
    cuda_path = "/usr/local/cuda"
    if os.path.exists(os.path.join(cuda_path, "bin/nvcc")):
        return cuda_path
    raise RuntimeError("Cannot find CUDA path")


def get_cuda_version(cuda_path=None):