INNER CODE UNIT · Python
get_cuda_driver_version
awslabs/awsome-distributed-ai · validation/efa-versions.py:53
def get_cuda_driver_version(container=[]):
try:
version = subprocess.check_output(container + ['nvidia-smi'])
version = version.decode('utf-8')
version = re.search(r'Driver Version: (\d+\.\d+\.\d+)', version).group(1)
return version
except Exception as e:
print(f'Error: {e}')
return None
def get_cuda_version(container=[]):
try:
version = subprocess.check_output(container + ['nvcc', '--version'])
version = version.decode('utf-8')
version = re.search(r', V(\d+\.\d+\.\d+)', version).group(1)
return version
except Exception as e:
print(f'Error: {e}')