torch.cuda.init#

torch.cuda.init()[source]#

Initialize PyTorch’s CUDA state.

You may need to call this explicitly if you are interacting withPyTorch via its C API, as Python bindings for CUDA functionalitywill not be available until this initialization takes place.Ordinary users should not need this, as all of PyTorch’s CUDA methodsautomatically initialize CUDA state on-demand.

Does nothing if the CUDA state is already initialized.