问题:
pip3 install torch==1.8.1 torchvision==0.9.1
Collecting torch==1.8.1
Downloading torch-1.8.1-cp38-cp38-manylinux1_x86_64.whl (804.1 MB)
|████████████████████████████████| 804.1 MB 183 kB/s eta 0:00:01Killed
如下,几乎下载到100%就被killed
问题原因:
根本原因就是虚拟机分配的内存不足以安装torch
解决方法:
1. 降低内存安装,在安装命令后面加上--no-cache-dir,如
参考:https://github.com/pytorch/pytorch/issues/1022
pip3 install torch==1.8.1 torchvision==0.9.1 --no-cache-dir
2. 先下载在安装
wget https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp38-cp38-linux_x86_64.whl
pip install 路径
3. 如果--no-cache-dir
标志是不够的,尝试增加交换空间
创建一个交换文件。
- 在当前文件系统上创建一个交换文件,例如在root上,为此可以创建一个新目录。
$ sudo mkdir /swap
- 在这个新目录中创建一个新文件,在本例中,创建一个2GB的新文件。
$ sudo dd if=/dev/zero of=/swap/swapfile1 bs=1M count=2048
- 在已创建的文件上创建一个新的交换区域。
$ sudo mkswap /swap/swapfile1
- 更改文件的权限。
$ sudo chmod 600 /swap/swapfile1
- 将交换分区添加到/etc/fSTAB文件中,如下所示,在此步骤中:
/swap/swapfile1 swap swap defaults 0 0
- 加载为实例创建的新交换空间。
sudo swapon -a
参考:3 Ways of Increasing Swap Space on Linux – The Geek Diary