无法在AWS EC2上安装PyTorch

Question

我创建了 EC2 实例（t2.micro、1 GiB RAM、30 GiB 磁盘）。 我连接到实例，然后成功安装 Flask 作为

pip3 install flask

但是当我尝试将 PyTorch 安装为

pip3 install torch

最后一刻失败了：

Collecting torch
  Downloading torch-2.2.1-cp310-cp310-manylinux1_x86_64.whl (755.5 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╸ 754.9/755.5 MB 32.8 MB/s eta 0:00:01Killed

Stackoverflow 社区，请施展你的魔法。谢谢你。

Answer 1

您的 RAM 内存不足。您可以通过运行

watch free -h

在另一个终端中检查它，同时在另一个终端中运行

pip install torch

。

要修复它，我建议您创建一个交换文件。交换文件充当额外的 RAM 内存，但由于它从硬盘驱动器运行，因此速度有点慢。但是，当您想要分配内存峰值时（就像您的情况一样），它们非常有用。以下示例展示了如何创建 5GB 交换文件：

sudo fallocate -l 1G ~/swapfile
sudo dd if=/dev/zero of=~/swapfile bs=1024 count=1048576
sudo chmod 600 ~/swapfile
sudo mkswap ~/swapfile
sudo swapon ~/swapfile

请记住，t2.micro 实例非常有限。请注意您所做的事情，并在需要时考虑升级到更大的实例。