Функция backends.cudnn.version

Функция backends.cudnn.version возвращает версию библиотеки cuDNN, с которой скомпилирован PyTorch. Это полезно для проверки совместимости и отладки производительности. Функция не принимает параметров и возвращает целое число, представляющее версию cuDNN.

Синтаксис

torch.backends.cudnn.version()

Пример

Давайте получим версию cuDNN, используемую в PyTorch:

import torch

ver = torch.backends.cudnn.version()
print(ver)

Результат выполнения кода (зависит от установленной версии):

8902

Пример

Проверим, доступен ли cuDNN, и получим его версию:

import torch

if torch.backends.cudnn.is_available():
	ver = torch.backends.cudnn.version()
	print(f"cuDNN version: {ver}")
else:
	print("cuDNN is not available")

Результат выполнения кода:

"cuDNN version: 8902"

Пример

Используем версию cuDNN для условной оптимизации кода:

import torch

ver = torch.backends.cudnn.version()
if ver >= 8000:
	torch.backends.cudnn.benchmark = True
	print("cuDNN 8.0+ detected, enabling benchmark")
else:
	print(f"cuDNN version {ver}, benchmark not enabled")

Результат выполнения кода:

"cuDNN 8.0+ detected, enabling benchmark"

Пример

Сравним версию cuDNN с минимально необходимой:

import torch

min_ver = 7600
cur_ver = torch.backends.cudnn.version()
if cur_ver >= min_ver:
	print("cuDNN version is sufficient")
else:
	print(f"cuDNN version {cur_ver} is below required {min_ver}")

Результат выполнения кода:

"cuDNN version is sufficient"

Смотрите также

функцию cudnn.is_available,
которая проверяет доступность cuDNN
функцию is_available,
которая проверяет доступность CUDA
функцию get_device_name,
которая возвращает имя GPU-устройства
функцию cudnn.benchmark,
которая включает автоматический подбор алгоритмов cuDNN