Skip to content

Commit 2b2be40

Browse files
committed
uv pip install
1 parent e15ddb7 commit 2b2be40

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

.lightning/workflows/fabric.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ timeout: "55" # minutes
88
parametrize:
99
matrix: {}
1010
include:
11-
# note that this is setting also all oldest requirements which is linked to Torch == 2.1
11+
# note that this is setting also all oldest requirements which is linked to python == 3.10
1212
- image: "nvidia/cuda:12.1.1-devel-ubuntu22.04"
1313
PACKAGE_NAME: "fabric"
1414
python_version: "3.10"
@@ -88,6 +88,8 @@ run: |
8888
uv pip install "cython<3.0" wheel # for compatibility
8989
fi
9090
91+
# install the base so we can adjust other packages
92+
uv pip install . --find-links="${TORCH_URL}"
9193
echo "Adjust torch versions in requirements files"
9294
PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
9395
uv pip install wget packaging
@@ -108,7 +110,7 @@ run: |
108110
fi
109111
110112
extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
111-
uv pip install -e ".[${extra}dev]" --upgrade --find-links="${TORCH_URL}"
113+
uv pip install ".[${extra}dev]" --upgrade --find-links="${TORCH_URL}"
112114
113115
python requirements/collect_env_details.py
114116
python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu >= 2, f'GPU: {mgpu}'"

0 commit comments

Comments
 (0)