cuda runtime error (2) : out of memory解决
cuda runtime error (2) : out of memory解决报错如图一般遇到out of memory的情况都是服务器的容器上在跑别的程序或是内存没释放,模型太大等,不过这次出现这个错误用nvidia-smi试了下,容器只用了10MB,模型也是不可能超内存的。后来查了下应该在dataloder中加入pin_memory=False,改完之后,程序立马跑通。...
原创
2020-08-22 15:55:28 ·
6844 阅读 ·
1 评论