Python教程

pytorch torch.save()方法保存数据被强制杀死进程，报错：killed 端口号 PYTHONUNBUFFERED=1

本文主要是介绍pytorch torch.save()方法保存数据被强制杀死进程，报错：killed 端口号 PYTHONUNBUFFERED=1，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

使用pytorch的来保存变量的时候，被Ubuntu系统强制杀死，其实原因主要就是因为内存不够，这是Ubuntu的一种自我保护机制，为了防止还加载大数据时内存不够而导致系统崩溃，所以采取的一种系统自我保护措施，我大概是保存了20个G的数据，用torch.save保存成pt文件，都加载完了，结果save的时候被杀死。
解决方法：
设置虚拟内存swap。其实目的就是为了在内存条的容量不够的时候，用硬盘上临时画出来的空间来代替内存条。当然，swap肯定是不如内存条读写快的，所以要费劲一些，但是可以解决内存爆掉的问题呢。
Ubuntu扩充swap的方法：
1、找个固定的位置，新建一个文件夹

mkdir swap

2、设置虚拟内存大小，count就是内存大小，单位是G，这个就是16G，执行成功的结果大概就是记录了啥啥的写入

sudo dd if=/dev/zero of=swapfile bs=1073741824 count=16

3、这步执行成功会输出一个UUID的号

sudo mkswap swapfile

4、最后一步，这一步执行时间较长，不要着急，耐心等待，

sudo swapon swapfile

5、成功后使用top命令测试下，swap那一项就是虚拟内存空间

top
这篇关于pytorch torch.save()方法保存数据被强制杀死进程，报错：killed 端口号 PYTHONUNBUFFERED=1的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航