From 0ab4b474c0e56a7c4bfda792640c9060623e3bd0 Mon Sep 17 00:00:00 2001 From: root Date: Tue, 14 Aug 2018 02:37:57 +0000 Subject: [PATCH] doc: update training with GPU parameter setting in maindir's README --- README.cn.md | 4 ++-- README.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.cn.md b/README.cn.md index 58730f8..ffef24c 100644 --- a/README.cn.md +++ b/README.cn.md @@ -51,12 +51,12 @@ nvidia-docker run -d -p 8888:8888 docker.paddlepaddlehub.com/book:latest-gpu 还需要将以下代码 ```python -paddle.init(use_gpu=False, trainer_count=1) +use_cuda = False ``` 改成： ```python -paddle.init(use_gpu=True, trainer_count=1) +use_cuda = True ``` diff --git a/README.md b/README.md index 6a18b6f..4c8d1c9 100644 --- a/README.md +++ b/README.md @@ -55,12 +55,12 @@ nvidia-docker run -d -p 8888:8888 docker.paddlepaddlehub.com/book:latest-gpu Change the code in the chapter that you are reading from ```python -paddle.init(use_gpu=False, trainer_count=1) +use_cuda = False ``` to: ```python -paddle.init(use_gpu=True, trainer_count=1) +use_cuda = True ``` -- GitLab