Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
5b99b598
Unverified
Commit
5b99b598
authored
Jul 14, 2021
by
Yuge Zhang
Committed by
GitHub
Jul 14, 2021
Browse files
Fix pipeline dependency (#3926)
parent
0c3827b3
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
21 additions
and
20 deletions
+21
-20
dependencies/recommended_gpu.txt
dependencies/recommended_gpu.txt
+13
-0
pipelines/full-test-linux.yml
pipelines/full-test-linux.yml
+4
-11
pipelines/full-test-windows.yml
pipelines/full-test-windows.yml
+4
-9
No files found.
dependencies/recommended_gpu.txt
0 → 100644
View file @
5b99b598
# Recommended because some non-commonly-used modules/examples depend on those packages.
-f https://download.pytorch.org/whl/torch_stable.html
tensorflow
keras
torch == 1.6.0
torchvision == 0.7.0
pytorch-lightning >= 1.1.1
onnx
peewee
graphviz
gym
tianshou >= 0.4.1
pipelines/full-test-linux.yml
View file @
5b99b598
...
...
@@ -15,8 +15,8 @@ jobs:
echo "##vso[task.setvariable variable=PATH]${PATH}:${HOME}/.local/bin"
echo "##vso[task.setvariable variable=NNI_RELEASE]999.$(date -u +%Y%m%d%H%M%S)"
python3 -m pip install -
-upgrade pip setuptools wheel
python3 -m pip install
pytes
t
python3 -m pip install -
U -r dependencies/setup.txt
python3 -m pip install
-r dependencies/develop.tx
t
displayName
:
Prepare
-
script
:
|
...
...
@@ -28,16 +28,9 @@ jobs:
-
script
:
|
set -e
python3 -m pip install scikit-learn==0.24.1
python3 -m pip install torchvision==0.7.0
python3 -m pip install torch==1.6.0
python3 -m pip install 'pytorch-lightning>=1.1.1'
python3 -m pip install keras==2.1.6
python3 -m pip install tensorflow==2.3.1 tensorflow-estimator==2.3.0
python3 -m pip install thop
python3 -m pip install pybnn
python3 -m pip install tianshou>=0.4.1 gym
sudo apt-get install swig -y
python3 -m pip install -r dependencies/recommended_gpu.txt
python3 -m pip install -e .[SMAC,BOHB,PPOTuner,DNGO]
displayName
:
Install extra dependencies
# Need del later
...
...
pipelines/full-test-windows.yml
View file @
5b99b598
...
...
@@ -12,8 +12,8 @@ jobs:
steps
:
-
script
:
|
python -m pip install -
-upgrade pip setuptools wheel
python -m pip install
pytes
t
python -m pip install -
U -r dependencies/setup.txt
python -m pip install
-r dependencies/develop.tx
t
displayName
:
Install Python tools
-
script
:
|
...
...
@@ -25,13 +25,8 @@ jobs:
displayName
:
Install NNI
-
script
:
|
python -m pip install scikit-learn==0.24.1
python -m pip install keras==2.1.6
python -m pip install torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
python -m pip install 'pytorch-lightning>=1.1.1'
python -m pip install tensorflow==2.3.1 tensorflow-estimator==2.3.0
python -m pip install pybnn
python -m pip install tianshou>=0.4.1 gym
python -m pip install -r dependencies/recommended.txt
python -m pip install -e .[DNGO]
displayName
:
Install extra dependencies
# Need del later
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment