Installing PyTorch with CUDA in setup.py

Question

I'm trying to specify PyTorch with CUDA in install_requires. The command to install with pip is

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

How do I do that in the setup.py install_requires?

You can specify the wheel file stackoverflow.com/a/59122860/6210807 — Harshit Kumar
– Harshit Kumar, Commented Mar 23, 2021 at 16:02

Hugh Perkins · Accepted Answer · 2022-12-02 18:48:20Z

4

I also faced same problem later I fixed it but using this in setup.py files and it worked, just add these lines as in your setup.py file.

"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"

All these are for linux version if you wantr any MACOSX or windows just change the link after @ in each line you can get link of your desired version from https://download.pytorch.org/whl/torch_stable.html

edited Dec 2, 2022 at 18:48

Hugh Perkins

8,7028 gold badges67 silver badges78 bronze badges

answered Oct 7, 2021 at 17:49

Vibhor Karnawat

1002 silver badges13 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Matt Pitkin · Accepted Answer · 2023-10-05 10:18:00Z

0

As of (the just released) PyTorch v2.1.0, you no longer have to specify the PyTorch CUDA-specific wheel (for Linux at least). If you can use CUDA 12.1, then you can just have "torch >= 2.1.0" as the dependency.

answered Oct 5, 2023 at 10:18

Matt Pitkin

7,3101 gold badge26 silver badges50 bronze badges

Collectives™ on Stack Overflow

Installing PyTorch with CUDA in setup.py

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related