ray/release/rllib_tests/regression_tests/run_tf.sh

12 lines
295 B
Bash
Raw Normal View History

#!/usr/bin/env bash
conda uninstall -y terminado
pip install -U pip
pip install "ray[rllib]"
pip install terminado
pip install tensorflow-gpu==2.3.0 # Change for new CUDA version
pip install boto3==1.4.8 cython==0.29.0
# Run tf learning tests.
rllib train -f compact-regression-tests-tf.yaml