diff --git a/.travis.yml b/.travis.yml index dec88f9f6..149c502cc 100644 --- a/.travis.yml +++ b/.travis.yml @@ -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 diff --git a/ci/travis/install-dependencies.sh b/ci/travis/install-dependencies.sh index 8504ee804..248e756f3 100755 --- a/ci/travis/install-dependencies.sh +++ b/ci/travis/install-dependencies.sh @@ -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