Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.azure/gpu-benchmarks.yml
View File

@ -75,7 +75,9 @@ jobs:
pip list pip list
displayName: "Image info & NVIDIA" 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: env:
FREEZE_REQUIREMENTS: "1" FREEZE_REQUIREMENTS: "1"
displayName: "Install package" displayName: "Install package"

1
.azure/gpu-tests-fabric.yml
View File

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

1
.azure/gpu-tests-pytorch.yml
View File

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