提交 1d46e843 编写于 作者: G guru4elephant

import os for serve.py

上级 620bdfd1
...@@ -63,6 +63,7 @@ def start_gpu_card_model(gpuid, args): # pylint: disable=doc-string-missing ...@@ -63,6 +63,7 @@ def start_gpu_card_model(gpuid, args): # pylint: disable=doc-string-missing
def start_multi_card(args): # pylint: disable=doc-string-missing def start_multi_card(args): # pylint: disable=doc-string-missing
gpus = "" gpus = ""
if args.gpu_ids == "": if args.gpu_ids == "":
import os
gpus = os.environ["CUDA_VISIBLE_DEVICES"] gpus = os.environ["CUDA_VISIBLE_DEVICES"]
else: else:
gpus = args.gpu_ids.split(",") gpus = args.gpu_ids.split(",")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册