用于GPU和GPU平台的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支持的电筒。

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

Answer 1

检查https://pytorch.org/。您将看到"MacOS二进制文件不支持CUDA，如果需要从源代码安装CUDA“。但是，您仍然可以通过安装MPS加速版本的py手电筒获得性能提升(这将取决于您的硬件)：

# MPS acceleration is available on MacOS 12.3+
pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu

这个命令可以在这里生成：https://pytorch.org/

​

​

为了在不同的平台上安装不同的火炬版本，您可以在requirements.txt中使用如下条件

# for CUDA 11.3 torch on Linux
--find-links https://download.pytorch.org/whl/cu113/torch_stable.html; sys_platform == "linux"
torch==1.10.2; sys_platform == "linux"
torchvision==0.11.3; sys_platform == "linux"
pytorch-lightning==1.5.10; sys_platform == "linux"

# for MPS accelerated torch on Mac
--pre --extra-index-url https://download.pytorch.org/whl/nightly/cpu; sys_platform == "darwin"
torch==1.10.2; sys_platform == "darwin"
torchvision==0.11.3; sys_platform == "darwin"
pytorch-lightning==1.5.10; sys_platform == "darwin"

# for CPU torch on Mac
# torch==1.10.2; sys_platform == "darwin"
# torchvision==0.11.3; sys_platform == "darwin"
# pytorch-lightning==1.5.10; sys_platform == "darwin"

这将在Linux上安装启用CUDA的火炬和火炬，但是MPS在MacOS上加速了它们的版本。