Skip to content

Commit f4b1ad8

Browse files
authored
deps: pin cuda-python >=12.0, <13.0.0 (#2410)
1 parent 5c18b33 commit f4b1ad8

2 files changed

Lines changed: 7 additions & 3 deletions

File tree

.azure/notebook-runs.yml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,11 @@ jobs:
5151
# drop pt from requirements so not to interfere with the existing one
5252
bash scripts/remove-torch-lines.sh requirements/base.txt
5353
cat requirements/base.txt
54-
# double check on test requirements
55-
pip install -U -r requirements/base.txt -r requirements/notebooks.txt
54+
pip install -U -r requirements/notebooks.txt
5655
# install this package
5756
python setup.py develop
57+
# double check on test requirements
58+
echo "Install special requirements for notebooks"
5859
displayName: "Install package & ..."
5960
6061
- bash: |

requirements/notebooks.txt

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
1+
# just reuse the base requirements
2+
-r base.txt
3+
14
ipython[all] ~=8.37.0
25
numpy
36
liger-kernel == 0.4.0
4-
cuda-python
7+
cuda-python >=12.0, <13.0.0 # todo
58
litgpt == 0.5.1

0 commit comments

Comments
 (0)