wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-11 11:24:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

[RLlib] Switch to PyTorch 1.6 for testing. (#9790)

Browse Source
This commit is contained in:
Sven Mika
2020-08-01 05:21:23 +02:00
committed by GitHub
parent 37a9c5783c
commit b4c527b3f3
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.travis.yml
+7 -7
View File
@@ -137,7 +137,7 @@ matrix:
- RLLIB_TESTING=1 RLLIB_REGRESSION_TESTS=1
- TF_VERSION=2.1.0
- TFP_VERSION=0.8
- TORCH_VERSION=1.4
- TORCH_VERSION=1.6
- PYTHON=3.6
- PYTHONWARNINGS=ignore
install:
@@ -154,7 +154,7 @@ matrix:
- RLLIB_TESTING=1 RLLIB_REGRESSION_TESTS_TF1X=1
- TF_VERSION=1.14.0
- TFP_VERSION=0.7
- TORCH_VERSION=1.4
- TORCH_VERSION=1.6
- PYTHON=3.6
- PYTHONWARNINGS=ignore
install:
@@ -170,7 +170,7 @@ matrix:
- RLLIB_TESTING=1 RLLIB_REGRESSION_TESTS_TORCH=1
- TF_VERSION=2.1.0
- TFP_VERSION=0.8
- TORCH_VERSION=1.4
- TORCH_VERSION=1.6
- PYTHON=3.6
- PYTHONWARNINGS=ignore
install:
@@ -188,7 +188,7 @@ matrix:
- PYTHON=3.6
- TF_VERSION=2.1.0
- TFP_VERSION=0.8
- TORCH_VERSION=1.4
- TORCH_VERSION=1.6
- PYTHONWARNINGS=ignore
install:
- . ./ci/travis/ci.sh init RAY_CI_RLLIB_FULL_AFFECTED
@@ -207,7 +207,7 @@ matrix:
- PYTHON=3.6
- TF_VERSION=2.1.0
- TFP_VERSION=0.8
- TORCH_VERSION=1.4
- TORCH_VERSION=1.6
- PYTHONWARNINGS=ignore
install:
- . ./ci/travis/ci.sh init RAY_CI_RLLIB_FULL_AFFECTED
@@ -226,7 +226,7 @@ matrix:
- PYTHON=3.6
- TF_VERSION=2.1.0
- TFP_VERSION=0.8
- TORCH_VERSION=1.4
- TORCH_VERSION=1.6
- PYTHONWARNINGS=ignore
install:
- . ./ci/travis/ci.sh init RAY_CI_RLLIB_FULL_AFFECTED
@@ -242,7 +242,7 @@ matrix:
- PYTHON=3.6
- TF_VERSION=2.1.0
- TFP_VERSION=0.8
- TORCH_VERSION=1.4
- TORCH_VERSION=1.6
- PYTHONWARNINGS=ignore
install:
- . ./ci/travis/ci.sh init RAY_CI_RLLIB_FULL_AFFECTED
ci/travis/install-dependencies.sh
+3 -3
View File
@@ -303,16 +303,16 @@ install_dependencies() {
pip install -r "${WORKSPACE_DIR}"/python/requirements_tune.txt
fi
# If CI has deemed that a different version of Tensorflow or Torch
# If CI has deemed that a different version of Tensorflow or Torch
# should be installed, then upgrade/downgrade to that specific version.
if [ -n "${TORCH_VERSION-}" ] || [ -n "${TFP_VERSION-}" ] || [ -n "${TF_VERSION-}" ]; then
case "${TORCH_VERSION-1.4}" in
case "${TORCH_VERSION-1.6}" in
1.5) TORCHVISION_VERSION=0.6.0;;
*) TORCHVISION_VERSION=0.5.0;;
esac
pip install --upgrade tensorflow-probability=="${TFP_VERSION-0.8}" \
torch=="${TORCH_VERSION-1.4}" torchvision=="${TORCHVISION_VERSION}" \
torch=="${TORCH_VERSION-1.6}" torchvision=="${TORCHVISION_VERSION}" \
tensorflow=="${TF_VERSION-2.2.0}" gym
fi