如何 setup.py 我依赖 PyTorch 的包

Question

我正在创建一个依赖于 PyTorch 的 Python 包。 PyTorch的安装命令如下（来自https://pytorch.org/）：

pip3 install torch==1.8.2+cu102 torchvision==0.9.2+cu102 torchaudio==0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

问题1：我应该如何准备设置.py？

我读了这个相当于`setup.py`中的`--find-links` 其中表示我们可以将链接添加到 dependency_links 列表中，但自 2019 年以来不再支持此功能。

问题 2：如何以编程方式决定要使用哪个版本 安装？ （CPU、GPU、CUDA 版本）

上面的命令适用于具有 CUDA 10.2 且支持 GPU 的计算机。但如果机器没有 GPU，人们会使用：

pip3 install torch==1.8.2+cpu torchvision==0.9.2+cpu torchaudio==0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

我的包的 setup.py 是否可以自动识别用户的系统并相应地修改 install_requires 列表？