一轮要跑几个小时
import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
cpu
原来是没用GPU
点击修改 -> 笔记本设置->硬件加速器改为GPU,保存即可
一轮要跑几个小时
import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
cpu
原来是没用GPU
点击修改 -> 笔记本设置->硬件加速器改为GPU,保存即可