原文:RuntimeError: "mse_cuda" not implemented for 'Long' 报错

criterion nn.MSELoss criterion a, b 这是a的dtype torch.float,b的dtype torch.int 因此,都改成float ...

2021-11-07 22:23 0 6243 推荐指数:

查看详情

存在空闲CUDA前提下报错RuntimeError: CUDA error: out of memory

问题背景: 最近跑代码时发现报错CUDA out of memory,进入linux终端查看GPU使用情况(nvidia-smi),结果如下: 我用的GPU序号是0,但这块被人占用了,所以我可以用剩下的3号和4号。 解决方案: 在代码中更改GPU使用序号(修改/添加代码 ...

Fri Sep 17 17:36:00 CST 2021 0 111
*** 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 ...

Wed Sep 15 05:24:00 CST 2021 0 1636
 
粤ICP备18138465号  © 2018-2026 CODEPRJ.COM