Skip to content

Commit bcc0714

Browse files
committed
fixed subprocess error
1 parent de01355 commit bcc0714

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nvdocker/nvdocker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def exec_run(self, cid, cmd):
153153
@staticmethod
154154
def list_gpus():
155155
#output = check_output(["nvidia-smi", "-L"]).decode("utf-8")
156-
query_gpu = check_output("nvidia-smi", "--query-gpu=memory.free,memory.used,memory.total --format=csv,noheader")
156+
query_gpu = check_output(["nvidia-smi", "--query-gpu=memory.free,memory.used,memory.total --format=csv,noheader"]).decode("utf-8");
157157
#regex = re.compile(r"GPU (?P<id>\d+):")
158158
gpus = {}
159159
id = 0;

0 commit comments

Comments
 (0)