深度学习
文章平均质量分 62
涛O_O
这个作者很懒,什么都没留下…
展开
-
构建含TensorFlow的树莓派docker镜像
Docker多架构构建设置 参考:Leverage multi-CPU architecture support 首先查看构建器builder: docker buildx ls 如果在支持的架构里面看到没有arm架构的版本,则先运行: docker run --rm --privileged multiarch/qemu-user-static --reset -p yes 然后创建自己的builder,并查看信息: docker buildx rm builder docker buildx cre原创 2021-04-12 13:09:27 · 458 阅读 · 0 评论 -
使用pytorch时 报错:OSError: [Errno 12] Cannot allocate memory
问题描述 在使用Pytorch跑模型时,虽然用了GPU,但随着训练的进行,后台占用内存(不是显存)越来越多,直到内存不足,程序崩溃: OSError: [Errno 12] Cannot allocate memory 解决办法 造成这个问题的原因是我在使用循环计算一个epoch内所有batch的loss之和时,直接累加了tensor,导致梯度信息一直保持,内存占用不断增加: for ep...原创 2019-10-17 12:20:47 · 3174 阅读 · 0 评论