AWS EC2 t2.micro实例pip安装PyTorch时内存耗尽求助

问题原因 #

t2.micro实例仅配置1GiB内存，PyTorch的安装包（755MB）在下载完成后，解压及后续安装过程会占用远超1GiB的内存，触发系统的OOM Killer（内存不足进程终止机制），直接杀掉安装进程，报错中的Killed就是明确信号。

解决步骤 #

• 临时添加交换分区（虚拟内存）补充内存容量：
  1. 创建2GiB的swap文件：

     sudo fallocate -l 2G /swapfile
     

  2. 限制swap文件权限（仅root可访问）：

     sudo chmod 600 /swapfile
     

  3. 格式化swap分区：

     sudo mkswap /swapfile
     

  4. 启用swap：

     sudo swapon /swapfile
     

  5. （可选）设置开机自动挂载swap：编辑/etc/fstab文件，添加以下内容后保存：

     /swapfile swap swap defaults 0 0
     

• 重新执行PyTorch安装命令：

  pip3 install torch
  

• （可选）安装完成后清理swap（仅当不需要虚拟内存时操作）：

  sudo swapoff /swapfile
  sudo rm /swapfile
  

  同时删除/etc/fstab中添加的swap挂载行。

内容的提问来源于stack exchange，提问作者Bob