Permalink
Browse files

round created_vram_gb to nearest value.

  • Loading branch information...
iperov committed Dec 6, 2018
1 parent 52f2be8 commit f90b2d231ba5b2e694a1e8e708e22c6c69578b4f
Showing with 1 addition and 1 deletion.
  1. +1 −1 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()

0 comments on commit f90b2d2

Please sign in to comment.