Установка PyTorch с CUDA в setup.py

Question

Я пытаюсь указать PyTorch с CUDA в install_requires. Команда для установки с помощью 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?

Answer 1

Я также столкнулся с той же проблемой позже, я исправил ее, но использовал это в файлах, и это сработало, просто добавьте эти строки, как в вашем setup.py файл.

      "torch@https://download.pytorch.org/whl/cu111/torch-1.8.0%2Bcu111-cp37-cp37m-linux_x86_64.whl",
"torchvision@https://download.pytorch.org/whl/cu111/torchvision-0.9.0%2Bcu111-cp37-cp37m-linux_x86_64.wh",
"torchaudio@https://download.pytorch.org/whl/torchaudio-0.8.0-cp36-cp36m-linux_x86_64.whl"

Все это для версии Linux, если вы хотите любой MACOSX или Windows, просто измените ссылку после @ в каждой строке, вы можете получить ссылку на нужную версию из https://download.pytorch.org/whl/torch_stable.html

Answer 2

Начиная с (только что выпущенной ) версии PyTorch 2.1.0 вам больше не нужно указывать колесо, специфичное для PyTorch CUDA (по крайней мере, для Linux). Если вы можете использовать CUDA 12.1, вы можете просто"torch >= 2.1.0"как зависимость.