()
| 58 | |
| 59 | _NVCC_VERSION = None |
| 60 | def nvcc_version(): |
| 61 | global _NVCC_VERSION |
| 62 | if _NVCC_VERSION is None: |
| 63 | import subprocess |
| 64 | |
| 65 | # Attempt to get NVCC version |
| 66 | result = subprocess.run(['nvcc', '--version'], capture_output=True) |
| 67 | if result.returncode != 0: |
| 68 | raise Exception('Unable to run `nvcc --version') |
| 69 | _NVCC_VERSION = str(result.stdout).split(" release ")[-1].split(",")[0] |
| 70 | return _NVCC_VERSION |
| 71 | |
| 72 | _CUDA_INSTALL_PATH = None |
| 73 | def cuda_install_path(): |
no test coverage detected