diff --git a/parl/remote/job.py b/parl/remote/job.py index e512a5238fb6ce662cf81d664556a6358c7a7014..8e8587afe9c228eecc1f9f20aed59ad3b060eb09 100644 --- a/parl/remote/job.py +++ b/parl/remote/job.py @@ -17,6 +17,7 @@ import compatible_trick import os os.environ['XPARL'] = 'True' +os.environ['CUDA_VISIBLE_DEVICES'] = '' import argparse import cloudpickle import pickle