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()