Функция cuda.is_initialized

Функция cuda.is_initialized проверяет, была ли инициализирована система CUDA в текущем процессе. Первым параметром функция принимает необязательный аргумент device, указывающий индекс устройства (по умолчанию используется текущее устройство). Возвращает значение типа bool.

Синтаксис

torch.cuda.is_initialized(device=None)

Пример

Проверим, инициализирована ли CUDA сразу после импорта библиотеки:

import torch

res = torch.cuda.is_initialized()
print(res)

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

False

Пример

После выполнения операции на GPU система инициализируется автоматически:

import torch

if torch.cuda.is_available():
	t = torch.tensor([1, 2, 3, 4, 5]).cuda()
	res = torch.cuda.is_initialized()
	print(res)

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

True

Пример

Функцию можно использовать для проверки состояния CUDA перед выполнением операций:

import torch

if torch.cuda.is_available():
	if not torch.cuda.is_initialized():
		t = torch.tensor([1, 2, 3]).cuda()
	res = torch.cuda.is_initialized()
	print(res)

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

True

Пример

Проверка инициализации для конкретного устройства:

import torch

if torch.cuda.is_available():
	t = torch.tensor([1, 2, 3, 4, 5]).cuda(0)
	res = torch.cuda.is_initialized(0)
	print(res)

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

True

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

функцию is_available,
которая проверяет доступность CUDA-устройств
функцию device_count,
которая возвращает количество доступных GPU
функцию current_device,
которая возвращает индекс текущего устройства
функцию synchronize,
которая синхронизирует все потоки на текущем устройстве