Pin setuptools for gpu builds

This commit is contained in:
Luca Antiga 2024-12-10 21:22:52 +01:00
parent e15ee2d6d1
commit a334fee572
3 changed files with 5 additions and 1 deletions

View File

@ -75,7 +75,9 @@ jobs:
pip list
displayName: "Image info & NVIDIA"
- bash: pip install -e .[dev] --find-links ${TORCH_URL}
- bash: |
pip install -e .[dev] --find-links ${TORCH_URL}
pip install setuptools==75.6.0
env:
FREEZE_REQUIREMENTS: "1"
displayName: "Install package"

View File

@ -107,6 +107,7 @@ jobs:
- bash: |
extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
pip install setuptools==75.6.0
displayName: "Install package & dependencies"
- bash: |

View File

@ -111,6 +111,7 @@ jobs:
- bash: |
extra=$(python -c "print({'lightning': 'pytorch-'}.get('$(PACKAGE_NAME)', ''))")
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
pip install setuptools==75.6.0
displayName: "Install package & dependencies"
- bash: pip uninstall -y lightning