fix multigpu extractor
This commit is contained in:
1 parent
9b08199e87
commit
d2d0dcb5e7
2 files changed
+2
-2
No files matched your search
+1
-1
@@ -102,7 +102,7 @@ class device:
|
||||
elif device.backend == "tensorflow":
|
||||
for dev in cuda_devices:
|
||||
if dev['total_mem'] >= totalmemsize_gb*1024*1024*1024:
|
||||
result.append (i)
|
||||
result.append ( dev['index'] )
|
||||
|
||||
return result
|
||||
|
||||
|
||||
Reference in new issue
Block a user