CPU和GPU平台的pytorch的requirements.txt

Question

我正在尝试创建一个requirements.txt来使用pytorch，但希望它能够在 GPU 和非 GPU 平台上工作。

我在我的 Linux GPU 系统上做了类似的事情：

--find-links https://download.pytorch.org/whl/cu113/torch_stable.html

torch==1.10.2+cu113
torchvision==0.11.3+cu113
pytorch-lightning==1.5.10

这工作正常，软件包已安装，我可以使用支持 GPU 的 pytorch。

torch我想知道如何为 mac 和非 GPU 用户修改此设置以安装和 的非 cuda 软件包torchvision？我需要维护单独的requirements.txt文件吗？

Answer 1

2024 年 2 月更新

检查https://pytorch.org/。您将看到“CUDA 在 MacOS 上不可用，请使用默认包”。但是，您仍然可以通过安装MPS加速版本的 pytorch来获得性能提升（这取决于您的硬件） ：

# MPS acceleration is available on MacOS 12.3+
pip3 install torch torchvision torchaudio

该命令可以在此处生成： https: //pytorch.org/

为了在不同的平台上安装不同的 Torch 版本，您可以requirements.txt像这样使用条件

# for CUDA 11.8 torch on Linux
--index-url https://download.pytorch.org/whl/cu118; sys_platform == "linux"
torch; sys_platform == "linux"
torchvision; sys_platform == "linux"
pytorch-lightning; sys_platform == "linux"

# for MPS accelerated torch on Mac
torch; sys_platform == "darwin"
torchvision; sys_platform == "darwin"
pytorch-lightning; sys_platform == "darwin"

这将在 Linux 上安装支持 CUDA 的 torch 和 torchvision，但在 MacOS 上安装它们的 MPS 加速版本