mirrors/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2025-10-31 12:50:59 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed a NVIDIA-SMI problem submitted by 찬우 전

Browse Source
This commit is contained in:
k4yt3x 2019-03-13 11:59:42 -04:00 committed by K4YT3X
parent 7f3a377ea8
commit a963e407e0
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
bin/video2x.py
View File

@ -108,9 +108,12 @@ def check_memory():
Avalon.warning('Nvidia-smi not available, skipping available memory check')
Avalon.warning('If you experience error \"cudaSuccess out of memory\", try reducing number of threads you\'re using')
else:
# "0" is GPU ID. Both waifu2x drivers use the first GPU available, therefore only 0 makes sense
gpu_memory_available = (GPUtil.getGPUs()[0].memoryTotal - GPUtil.getGPUs()[0].memoryUsed) / 1024
memory_status.append(('GPU', gpu_memory_available))
try:
# "0" is GPU ID. Both waifu2x drivers use the first GPU available, therefore only 0 makes sense
gpu_memory_available = (GPUtil.getGPUs()[0].memoryTotal - GPUtil.getGPUs()[0].memoryUsed) / 1024
memory_status.append(('GPU', gpu_memory_available))
except ValueError:
pass
# Go though each checkable memory type and check availability
for memory_type, memory_available in memory_status: