torch.cuda.init

torch.cuda.init() [source]

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

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