*** RuntimeError: CUDA error: out of memory.

相关文章推荐

酷酷的鸵鸟 · GitHub Pages 与 Gitee ...· 1 年前 ·

深沉的鸡蛋面 · TypeScript：如何将数组转换为对象？· 1 年前 ·

豁达的跑步鞋 · lstm单步预测python ...· 2 年前 ·

年轻有为的墨镜 · mongoose中connect()、cre ...· 2 年前 ·

眼睛小的足球 · 结合实例学习 Typescript - 掘金· 2 年前 ·

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.

这个error 的原因是，当期指定的GPU的显存不足，可以关闭现有的process，或者重指定显卡编号。

device = torch.device("cuda:0")

推荐文章

酷酷的鸵鸟 · GitHub Pages 与 Gitee Pages 上的 Jekyll - DevOps在路上 - 博客园

1 年前

深沉的鸡蛋面 · TypeScript：如何将数组转换为对象？

1 年前

豁达的跑步鞋 · lstm单步预测python lstm预测原理_mob6454cc6d3e23的技术博客_51CTO博客

2 年前

年轻有为的墨镜 · mongoose中connect()、createConnection()和connection的区别和作用_mongoose.connect_younglao的博客-CSDN博客

2 年前

眼睛小的足球 · 结合实例学习 Typescript - 掘金

2 年前