round created_vram_gb to nearest value.

2018-12-06 09:32:28 +04:00
parent 52f2be80d7
commit f90b2d231b
1 changed files with 1 additions and 1 deletions
@@ -204,7 +204,7 @@ def getDeviceVRAMTotalGb (idx):
        memInfo = nvmlDeviceGetMemoryInfo( handle )
        result = memInfo.total / (1024*1024*1024)
    nvmlShutdown()
-    return result
+    return round(result)
 def getBestDeviceIdx():
    nvmlInit()