今天跑深度学习代码的时候遇到了报错
这个错误显示的并不是真正的报错,把代码从gpu改到cpu就可以显示真正的错误原因了
修改
device = torch.device(f'cuda:{cfg.gpu}' if torch.cuda.is_available() else 'cpu')
# 直接指定cpu
device = torch.device(f'cpu')
这样再运行就会显示真正的报错了,再修改就好了
今天跑深度学习代码的时候遇到了报错
这个错误显示的并不是真正的报错,把代码从gpu改到cpu就可以显示真正的错误原因了
修改
device = torch.device(f'cuda:{cfg.gpu}' if torch.cuda.is_available() else 'cpu')
# 直接指定cpu
device = torch.device(f'cpu')
这样再运行就会显示真正的报错了,再修改就好了