提交 d4c0517f 编写于 作者: R ronny1996

fix bug for rocm

上级 c81baca7
......@@ -311,7 +311,10 @@ def create_predictor(args, mode, logger):
def get_infer_gpuid():
if not paddle.fluid.core.is_compiled_with_rocm():
cmd = "env | grep CUDA_VISIBLE_DEVICES"
else:
cmd = "env | grep HIP_VISIBLE_DEVICES"
env_cuda = os.popen(cmd).readlines()
if len(env_cuda) == 0:
return 0
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册