From f90b2d231ba5b2e694a1e8e708e22c6c69578b4f Mon Sep 17 00:00:00 2001
From: iperov <lepersorium@gmail.com>
Date: Thu, 6 Dec 2018 09:32:28 +0400
Subject: [PATCH] round created_vram_gb to nearest value.

---
 gpufmkmgr/gpufmkmgr.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpufmkmgr/gpufmkmgr.py b/gpufmkmgr/gpufmkmgr.py
index a0dbb87..a8c0923 100644
--- a/gpufmkmgr/gpufmkmgr.py
+++ b/gpufmkmgr/gpufmkmgr.py
@@ -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()