Skip to content

Commit 6668586

Browse files
committed
renamed gpu_info function
1 parent 73ce1f4 commit 6668586

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
@@ -151,7 +151,7 @@ def exec_run(self, cid, cmd):
151151
return c.exec_run(cmd)
152152

153153
@staticmethod
154-
def get_gpus():
154+
def gpu_info():
155155
#output = check_output(["nvidia-smi", "-L"]).decode("utf-8")
156156
keys = ['memory_free', 'memory_used', 'memory_total']
157157
query_gpu = check_output(["nvidia-smi", "--query-gpu=memory.free,memory.used,memory.total","--format=csv,noheader"]).decode("utf-8")

0 commit comments

Comments
 (0)