diff --git a/03_nc_to_npy_for_openvdb.py b/03_nc_to_npy_for_openvdb.py index 9c21100..fcba4e3 100644 --- a/03_nc_to_npy_for_openvdb.py +++ b/03_nc_to_npy_for_openvdb.py @@ -50,12 +50,12 @@ def str2bool(v): else: raise argparse.ArgumentTypeError('Boolean value expected.') -def find_free_GPU_memory(gpu_id): +def find_free_GPU_memory(gpu_id, limit=0.5): free = cp.cuda.Device(gpu_id).mem_info[0]/cp.cuda.Device(gpu_id).mem_info[1] - while free<0.25: + while free