如何安装没有 GPU 依赖的 pytorch？

Question

我有一个项目依赖于torch==2.0.1，但它将在CPU而不是GPU上运行。 torch 有一些大的 cuda/cublas/cudnn 依赖项，我相信只有在 GPU 上运行时才需要这些依赖项。这些软件包（我认为我不需要）占了约 4.5 GB x 约 6.5 GB 映像。

如何在不安装 GPU 特定依赖项的情况下安装 torch？我正在使用诗歌 1.8.2 在我的

Linux 6.6.22-linuxkit x86_64

中安装软件包。

在运行

ARG CUDA_VISIBLE_DEVICES=""

之前，我在 Dockerfile 中尝试了

poetry install

希望这些依赖项不会被安装，但这不起作用。我知道 env 变量更多的是一个运行时的东西，告诉 torch 使用 CPU，但软件包已经由运行时安装了。我想我只是尝试将它设置在我的 Dockerfile 中只是为了好玩。

Answer 1

从 pytorch 网站 - 您可以使用以下命令安装仅 CPU 版本：

pip3 install torch torchvision torchaudio

或

conda install pytorch torchvision torchaudio cpuonly -c pytorch