From f5e1516907ecc18ac3f1a7bfae1c7b2ac4193df2 Mon Sep 17 00:00:00 2001 From: TomorrowIsAnOtherDay <2466956298@qq.com> Date: Wed, 19 Aug 2020 20:22:30 +0800 Subject: [PATCH] remove the default setting that unables GPUs in the job node --- parl/remote/job.py | 1 - parl/remote/tests/paddle_gpu_test.py | 1 - 2 files changed, 2 deletions(-) diff --git a/parl/remote/job.py b/parl/remote/job.py index 4f4579b..e512a52 100644 --- a/parl/remote/job.py +++ b/parl/remote/job.py @@ -16,7 +16,6 @@ import compatible_trick import os -os.environ['CUDA_VISIBLE_DEVICES'] = '' os.environ['XPARL'] = 'True' import argparse import cloudpickle diff --git a/parl/remote/tests/paddle_gpu_test.py b/parl/remote/tests/paddle_gpu_test.py index 8b759c7..256814c 100644 --- a/parl/remote/tests/paddle_gpu_test.py +++ b/parl/remote/tests/paddle_gpu_test.py @@ -29,7 +29,6 @@ import os @parl.remote_class class Actor(object): def __init__(self): - os.environ['CUDA_VISIBLE_DEVICES'] = "0" if parl.utils.is_gpu_available(): place = fluid.CUDAPlace(0) else: -- GitLab