File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -8,7 +8,7 @@ timeout: "55" # minutes
88parametrize :
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"
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}'"
You can’t perform that action at this time.
0 commit comments