github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 21:42:08 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

round created_vram_gb to nearest value.

Browse source
This commit is contained in:
iperov 2018-12-06 09:32:28 +04:00
parent 52f2be80d7
commit f90b2d231b
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
gpufmkmgr/gpufmkmgr.py
View file

@ -204,7 +204,7 @@ def getDeviceVRAMTotalGb (idx):
memInfo = nvmlDeviceGetMemoryInfo( handle ) memInfo = nvmlDeviceGetMemoryInfo( handle )
result = memInfo.total / (1024*1024*1024) result = memInfo.total / (1024*1024*1024)
nvmlShutdown() nvmlShutdown()
return result return round(result)
def getBestDeviceIdx(): def getBestDeviceIdx():
nvmlInit() nvmlInit()