Unverified Commit a9897ec0 authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Refactor integration test (step 9) - dependency version updates (#4900)

parent 8cc29b37
......@@ -43,9 +43,9 @@ RUN python3 -m pip --no-cache-dir install \
torchvision==0.11.3+cu113 \
torchaudio==0.10.2+cu113 \
-f https://download.pytorch.org/whl/cu113/torch_stable.html
RUN python3 -m pip --no-cache-dir install pytorch-lightning==1.5.10
RUN python3 -m pip --no-cache-dir install pytorch-lightning==1.6.1
RUN python3 -m pip --no-cache-dir install tensorflow==2.8.0
RUN python3 -m pip --no-cache-dir install tensorflow==2.9.1
RUN python3 -m pip --no-cache-dir install azureml==0.2.7 azureml-sdk==1.38.0
......
......@@ -2,8 +2,8 @@
-f https://download.pytorch.org/whl/torch_stable.html
tensorflow
torch == 1.10.0+cu111
torchvision == 0.11.1+cu111
torch == 1.10.0+cu113
torchvision == 0.11.1+cu113
pytorch-lightning >= 1.6.1
lightgbm
onnx
......
......@@ -14,9 +14,9 @@ graphviz
gym
tianshou >= 0.4.1
matplotlib
timm >= 0.5.4
# TODO: time to drop tensorflow 1.x
keras
tensorflow < 2.0
protobuf <= 3.20.1
timm >= 0.5.4
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment