« Minimal Reproducible »
Home Tags About GitHub

Posts with tag pip

Specify cuda version in setup.py when pip install -e .

2024-11-26
installlangspippython

Specify cuda version in setup.py when pip install -e .see: https://stackoverflow.com/questions/66738473/installing-pytorch-with-cuda-in-setup-pyIn setup function, use something like:"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"version links are from: https://download.pytorch.org/whl/torch_stable.htmlmethod above is ok to download, but still failed to build.The following succeeds for building sam-2 with cu118:pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu118 pip install --no-build-isolation -e . # and delete these packages in `setup.py

No more posts to load.