python-3.x - 如何在没有GPU支持的情况下编译torch 1.5.0？

Question

我想在 AWS lambda 上安装 pytorch 1.5.0。由于 torch 库非常大，我需要使其尽可能小以适应大小限制。到目前为止，我的脚本如下所示:

    mkdir python
    docker run \
        --rm \
        -v $(pwd):/build \
        python:3.8 \
        sh -c "
            cd /build;
            pip3 install torch==1.5.0 -t python/torch --no-cache-dir;
            find . -type d -name '__pycache__' | xargs rm -rf;
            find . -type d -name 'tests' | xargs rm -rf;
            find . -type f -name '*.py[co]' | xargs rm -rf;
        ";

    zip -r9 torch.zip python;

但是生成的 zip 文件非常大(500+ MB)。然而，安装包中最大的文件之一是libtorch_cuda.so。删除该文件会使 zip 文件的大小减少一半。我知道cuda是GPU的库，并且由于AWS lambda没有GPU，所以我不需要这种支持。但是当我删除该文件时， torch 将无法正确导入。

相比之下，torch 1.4.0 要小得多，因为它默认不包含 cuda 库。

我想要没有 GPU 支持的 torch 1.5.0。

有没有办法在没有 GPU 支持的情况下 pip install torch==1.5.0 ？

Answer 1

PyTorch 还分发仅限 CPU 的版本，您可以使用 pip 安装。虽然它们没有发布到 PyPI，所以您需要从它们自己的注册表中获取它们。

您可以在PyTorch - Getting Started Locally获取CPU版本通过选择 CUDA:无。

pip install torch==1.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html