device = next(model.parameters()).device print(f"Model is on: {device}")
is_cuda = next(model.parameters()).is_cuda if is_cuda: print("Model is on GPU") else: print("Model is on CPU")
上一篇:Unity TMP(TextMesh Pro)的中文输入相关问题
下一篇:Spring Cache框架详解