Skip to content

Commit 4a17016

Browse files
committed
REvert CPU Versions
1 parent 8f26115 commit 4a17016

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
numpy
22
-f https://download.pytorch.org/whl/torch_stable.html
3-
torch>=2.6.0+cpu
3+
torch==2.0.1+cpu

tox.ini

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,8 +83,8 @@ passenv =
8383
commands =
8484
python -c "import os; os.system('install-custom-pkgs --install-boto-wheels')"
8585
pip install 'apache-airflow==2.9.3' --constraint "https://raw.githubusercontent.com/apache/airflow/constraints-2.9.3/constraints-3.8.txt"
86-
pip install 'torch>=2.6.0+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
87-
pip install 'torchvision>=0.17.0+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
86+
pip install 'torch==2.0.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
87+
pip install 'torchvision==0.15.2+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
8888
pip install 'dill>=0.3.8'
8989

9090
pytest {posargs}

0 commit comments

Comments
 (0)