torch.cuda.is_initialized

torch.cuda.is_initialized()

返回 PyTorch 的 CUDA 狀態是否已初始化。