当前位置:   article > 正文

cuda runtime error (2) : out of memory解决_program hit out of memory (error 2) on cuda api ca

program hit out of memory (error 2) on cuda api call to cudastreamcreatewith

cuda runtime error (2) : out of memory解决

在这里插入图片描述
报错如图
一般遇到out of memory的情况都是服务器的容器上在跑别的程序或是内存没释放,模型太大等,不过这次出现这个错误用nvidia-smi试了下,容器只用了10MB,模型也是不可能超内存的。后来查了下应该在DataLoader中加入pin_memory=False,改完之后,程序立马跑通。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/220571
推荐阅读
相关标签
  

闽ICP备14008679号