在 setup.py 中使用 CUDA 安装 PyTorch

问题描述

我正在尝试在 install_requires 中指定带有 CUDA 的 PyTorch。用pip安装的命令是

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

如何在 setup.py install_requires 中执行此操作？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）