Use the 65 percentile of free_mem for GPU selection, instead of 75 percentile

This commit is contained in:
cmdr2 2022-11-14 12:26:21 +05:30
parent d79eb5e1a6
commit 027b2e1b88

View File

@ -3,7 +3,7 @@ import torch
import traceback
import re
COMPARABLE_GPU_PERCENTILE = 0.75 # if a GPU's free_mem is within this % of the GPU with the most free_mem, it will be picked
COMPARABLE_GPU_PERCENTILE = 0.65 # if a GPU's free_mem is within this % of the GPU with the most free_mem, it will be picked
def get_device_delta(render_devices, active_devices):
'''