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

相关文章推荐

正直的豆芽 · [MY-010273] [Server] ...· 7 月前 ·

善良的鞭炮 · connection aborted.' ...· 1 年前 ·

爱听歌的刺猬 · Python json去除反斜杠 ...· 1 年前 ·

老实的稀饭 · recession - OSCHINA - ...· 1 年前 ·

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")

推荐文章

正直的豆芽 · [MY-010273] [Server] Could not create unix socket lock file /var/run/mysqld/mysqld.sock.lock.问题解决_my

7 月前

善良的鞭炮 · connection aborted.' badstatusline-掘金

1 年前

爱听歌的刺猬 · Python json去除反斜杠 python删除json某个内容_clghxq的技术博客_51CTO博客

1 年前

老实的稀饭 · recession - OSCHINA - 中文开源技术交流社区

1 年前