6

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?

talonmies
  • 70,661
  • 34
  • 192
  • 269
user3424455
  • 105
  • 1
  • 8

1 Answers1

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

Hugh Perkins
  • 7,975
  • 7
  • 63
  • 71
Vibhor Karnawat
  • 100
  • 1
  • 12