2021年9月14日
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
摘要： 在代码中设置指定GPU，代码仍旧运行不成功 device = torch.device(“cuda:3” if torch.cuda.is_available() else “cpu”)model = model.to(device) 在上面的代码前面添加了代码torch.cuda.set_devi 阅读全文
posted @ 2021-09-14 23:21 呦呦南山 阅读(3739) 评论(2) 推荐(1) 编辑
*** RuntimeError: CUDA error: out of memory.
摘要： CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing 阅读全文
posted @ 2021-09-14 21:24 呦呦南山 阅读(15991) 评论(0) 推荐(1) 编辑