diff --git a/parl/remote/job.py b/parl/remote/job.py index 4f4579b236679b0774129ad6764a035a0d92fdf1..e512a5238fb6ce662cf81d664556a6358c7a7014 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 8b759c7ab5f2aa87b64861b8f5d773fa509aa27d..256814c24ad533b107308d04b6f4b13d5291b3b1 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: