device = “cuda“ if torch.cuda.is_available() else “cpu“ model = model.to(device)不能使用GPU
device = "cuda" if torch.cuda.is_available() else "cpu"model = model.to(device)不能使用GPUPyCharmPyTorch神经网络CUDAtorch.cuda.is_available()to(device)
原创
2025-03-21 21:12:44 ·
461 阅读 ·
0 评论