wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-20 12:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

[rllib/tune] Add test for fractional gpu support in xray mode; add rllib support for fractional gpu (#2768)

Browse Source 
* frac gpu

* doc

* Update rllib-training.rst

* yapf

* remove xray
This commit is contained in:
Eric Liang
2018-09-03 11:12:23 -07:00
committed by GitHub
parent 9d655721e5
commit df4788e501
10 changed files with 41 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/jenkins_tests/run_multi_node_tests.sh
+4 -2
View File
@@ -289,14 +289,16 @@ docker run -e "RAY_USE_XRAY=1" --rm --shm-size=10G --memory=10G $DOCKER_SHA \
docker run -e "RAY_USE_XRAY=1" --rm --shm-size=10G --memory=10G $DOCKER_SHA \
python /ray/python/ray/rllib/examples/multiagent_two_trainers.py --num-iters=2
docker run -e "RAY_USE_XRAY=1" --rm --shm-size=10G --memory=10G $DOCKER_SHA \
# No Xray for PyTorch
docker run --rm --shm-size=10G --memory=10G $DOCKER_SHA \
python /ray/python/ray/rllib/train.py \
--env PongDeterministic-v4 \
--run A3C \
--stop '{"training_iteration": 2}' \
--config '{"num_workers": 2, "use_pytorch": true, "model": {"use_lstm": false, "grayscale": true, "zero_mean": false, "dim": 84, "channel_major": true}, "preprocessor_pref": "rllib"}'
docker run -e "RAY_USE_XRAY=1" --rm --shm-size=10G --memory=10G $DOCKER_SHA \
# No Xray for PyTorch
docker run --rm --shm-size=10G --memory=10G $DOCKER_SHA \
python /ray/python/ray/rllib/train.py \
--env CartPole-v1 \
--run A3C \