Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

Commit

Permalink
[CI] Fast-test cross-platform speedup (#3364)
Browse files Browse the repository at this point in the history
  • Loading branch information
ultmaster authored Feb 9, 2021
1 parent bb3d298 commit 65a2ceb
Show file tree
Hide file tree
Showing 5 changed files with 161 additions and 104 deletions.
29 changes: 10 additions & 19 deletions dependencies/recommended.txt
Original file line number Diff line number Diff line change
@@ -1,23 +1,14 @@
# Recommended because some non-commonly-used modules/examples depend on those packages.

-f https://download.pytorch.org/whl/torch_stable.html
tensorflow
torch >= 1.6+cpu, != 1.7+cpu -f https://download.pytorch.org/whl/torch_stable.html
torchvision >= 0.8+cpu -f https://download.pytorch.org/whl/torch_stable.html

# PyTorch 1.7 has compatibility issue with model compression.
# Check for MacOS because this file is used on all platforms.
torch == 1.6.0+cpu ; sys_platform != "darwin"
torch == 1.6.0 ; sys_platform == "darwin"
torchvision == 0.7.0+cpu ; sys_platform != "darwin"
torchvision == 0.7.0 ; sys_platform == "darwin"
onnx
peewee
thop
graphviz
tensorflow

# the following content will be read by setup.py.
# please follow the logic in setup.py.

# SMAC
ConfigSpaceNNI
smac4nni

# BOHB
ConfigSpace==0.4.7
statsmodels==0.12.0

# PPOTuner
enum34
gym
8 changes: 8 additions & 0 deletions dependencies/recommended_legacy.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
-f https://download.pytorch.org/whl/torch_stable.html
tensorflow == 1.15.4
torch == 1.5.1+cpu
torchvision == 0.6.1+cpu
keras == 2.1.6
onnx
peewee
graphviz
13 changes: 13 additions & 0 deletions dependencies/required_extra.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# the following content will be read by setup.py.
# please follow the logic in setup.py.

# SMAC
ConfigSpaceNNI
smac4nni

# BOHB
ConfigSpace==0.4.7
statsmodels==0.12.0

# PPOTuner
gym
Loading

0 comments on commit 65a2ceb

Please sign in to comment.