赞
踩
突然有一次运行就出现这个错误
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 CUDA_LAUNCH_BLOCKING=1.
搜了半天,看到有个大佬
这篇文章说:
可是我明明前几天刚用过啊,啥也没改就突然报错了,怎么想也不可能是机器突然显存不行了吧。而且我喵喵的就是用的学校该死的服务器!!!
然后,我在这个网站:https://www.cnblogs.com/naive-LR/p/14256624.html找到了解决办法,我哭了家人们,太搞我心态了。
就是在文件里加入“torch.backends.cudnn.enabled=False”这行代码。
torch.backends.cudnn.enabled=False
反正我加到了下图这个位置
这个指令其实加到哪里都行,主要是关闭cudnn加速的。
不过我感觉关了也没感觉怎么慢。
下图就是历经千辛万苦我终于能运行了。纪念一下
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。