diff --git a/.teamcity/Dockerfile b/.teamcity/Dockerfile index 85d8a2f37e0db3472d1ad02ab06fb7a80fd72c3a..e5422e8ec25afdd842dee657cce38c0f15b59b57 100644 --- a/.teamcity/Dockerfile +++ b/.teamcity/Dockerfile @@ -15,7 +15,7 @@ # A dev image based on paddle production image -FROM parl/parl-test:cuda9.0-cudnn7 +FROM parl/parl-test:1.1-cuda9.0-cudnn7 COPY ./requirements.txt /root/ diff --git a/.teamcity/build.sh b/.teamcity/build.sh index 34092a605f01fd5d11040ba30de7b2c84ba4ca09..9b8b7a3e7f9f8b0b925399c0938feb117e41b097 100755 --- a/.teamcity/build.sh +++ b/.teamcity/build.sh @@ -56,6 +56,8 @@ function check_style() { } function run_test_with_gpu() { + export FLAGS_fraction_of_gpu_memory_to_use=0.5 + mkdir -p ${REPO_ROOT}/build cd ${REPO_ROOT}/build cmake .. diff --git a/README.md b/README.md index 4e7a956a08e592c2e9612c8ad37f15631483b228..971f91867a4c0603871b0b6fc8b894497d848123 100644 --- a/README.md +++ b/README.md @@ -69,7 +69,7 @@ agent = AtariAgent(algorithm) # Parallelization PARL provides a compact API for distributed training, allowing users to transfer the code into a parallelized version by simply adding a decorator. -Here is a `Hello World` example to demonstrate how easily it is to leverage outer computation resources. +Here is a `Hello World` example to demonstrate how easy it is to leverage outer computation resources. ```python #============Agent.py================= @parl.remote_class