torch.cuda.init

torch.cuda.init() [исходный код]

Инициализирует состояние CUDA в PyTorch.

Вам может потребоваться вызвать эту функцию явно, если вы взаимодействуете с PyTorch через его C API, поскольку привязки Python для функций CUDA будут недоступны до завершения этой инициализации. Обычным пользователям это не требуется, поскольку все методы CUDA в PyTorch автоматически инициализируют состояние CUDA по мере необходимости.

Ничего не делает, если состояние CUDA уже инициализировано.