torch.cuda.is_initialized

torch.cuda.is_initialized()[source][source]

返回 PyTorch 的 CUDA 状态是否已初始化。