提交 f5e15169 编写于 作者: T TomorrowIsAnOtherDay

remove the default setting that unables GPUs in the job node

上级 af882e17
......@@ -16,7 +16,6 @@
import compatible_trick
import os
os.environ['CUDA_VISIBLE_DEVICES'] = ''
os.environ['XPARL'] = 'True'
import argparse
import cloudpickle
......
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册