-
-
Notifications
You must be signed in to change notification settings - Fork 715
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
memory leak when using distributed.Client with delayed #2068
Comments
I am seeing the same memory behavior with Python 3.5.5 and the above sample.
|
Interesting. When I run this I get something similar. Memory use climbs slowly in steps. I also get a number of warnings about garbage collection time taking a long while.
I'm curious how people generally debug this sort of issue. I might start with the following:
If anyone has any experience here and has the time to investigate this further I would appreciate it. |
tracemalloc may be an option: https://docs.python.org/3/library/tracemalloc.html#compute-differences |
There is also objgraph which is useful for generating reference graphs of objects: https://mg.pov.lt/objgraph/ |
It would be great if people could try these tools and report back what they
find
…On Thu, Jun 28, 2018 at 7:32 AM, Michael Peleshenko < ***@***.***> wrote:
There is also objgraph which is useful for generating reference graphs of
objects: https://mg.pov.lt/objgraph/
—
You are receiving this because you commented.
Reply to this email directly, view it on GitHub
<#2068 (comment)>,
or mute the thread
<https://github.com/notifications/unsubscribe-auth/AASszLJbYCyfBZvSubPrwnvgTasnqgrEks5uBL69gaJpZM4UzTGE>
.
|
Code for: mleak.pyimport gc
import os
import tracemalloc
import psutil
from dask import delayed
from dask.distributed import get_worker
# generate random strings: https://stackoverflow.com/a/16310739
class Data():
def __init__(self):
self.tbl = bytes.maketrans(bytearray(range(256)),
bytearray([ord(b'a') + b % 26 for b in range(256)]))
@staticmethod
def split_len(seq, length):
return [seq[i:i + length] for i in range(0, len(seq), length)]
def get_data(self):
l = self.split_len(os.urandom(1000000).translate(self.tbl), 1000)
return l
class Calc():
def __init__(self, l):
self.l = l
def nth_nth_item(self, n):
return self.l[n][n]
class Combiner():
def __init__(self):
self.delayed_data = delayed(Data())
def get_calc(self):
d_l = self.delayed_data.get_data(pure=True)
return delayed(Calc, pure=True)(d_l)
def mem_usage_mb(self):
process = psutil.Process(os.getpid())
return "%.2f" % (process.memory_info().rss * 1e-6)
def results(self):
return {
'0': self.get_calc().nth_nth_item(0),
'1': self.get_calc().nth_nth_item(1),
'2': self.get_calc().nth_nth_item(2),
'mem_usage_mb': self.mem_usage_mb()
}
def delayed_results(self):
return delayed(self.results())
##
##
def snapshot():
worker = get_worker()
worker.snapshot1 = tracemalloc.take_snapshot()
def top_stats(do_print_trace=False):
worker = get_worker()
snapshot2 = tracemalloc.take_snapshot()
stats = snapshot2.compare_to(worker.snapshot1, 'traceback')
for stat in stats[:5]:
print(stat)
if do_print_trace:
for line in stat.traceback.format():
print(line)
##
##
def main_threaded_get():
from dask.threaded import get as threaded_get
from dask import compute
for i in range(300):
delayed_obj = Combiner().delayed_results()
res = compute(delayed_obj, key=threaded_get)[0]
print("#%d, mem: %s mb" % (i, res['mem_usage_mb']))
gc.collect()
def main_distributed_client():
from distributed import Client
client = Client(processes=True, n_workers=1, threads_per_worker=1)
# up to 7 stacktrace lines
client.run(lambda: tracemalloc.start(7))
for i in range(1000):
client.run(lambda: snapshot())
delayed_obj = Combiner().delayed_results()
future = client.compute(delayed_obj)
res = future.result()
print("#%d, mem: %s mb" % (i, res['mem_usage_mb']))
client.run(lambda: top_stats(do_print_trace=False))
# client.run(lambda: top_stats(do_print_trace=True)) # print call stack as well
client.run(lambda: gc.collect()) # doesn't help
print("---")
if __name__ == "__main__":
main_distributed_client() Modified script compares memory usage using tracemalloc before computing delayed function and after. If I'm interpreting tracemalloc results correctly, then it looks that memory usage grows when pickle.loads is called. Run: Top memory increases per invocation:
Call stack for distributed/protocol/pickle.py (different invocation)
|
Hello, I have the same Issue. I tried to manually perform a client.restart() but the workers are restarting with theirs saturated memories. Is it possible to reset all workers memory? (already tried client.cancel and del xx, without success). Thanks |
but the workers are restarting with theirs saturated memories
What does this mean exactly?
Calling client.restart kills the worker process and then brings it back.
It's a pretty clean state.
…On Fri, Aug 31, 2018 at 2:08 AM, Axel CHERUBIN ***@***.***> wrote:
Hello, I have the same Issue. I tried to manually perform a
client.restart() but the workers are restarting with theirs saturated
memories. Is it possible to reset all workers memory? (already tried
client.cancel and del xx, without success). Thanks
—
You are receiving this because you commented.
Reply to this email directly, view it on GitHub
<#2068 (comment)>,
or mute the thread
<https://github.com/notifications/unsubscribe-auth/AASszDpgTnGpezeF6yIxZ1SoqsENJeZAks5uWNLbgaJpZM4UzTGE>
.
|
@mrocklin Thank you for your support. The contextI'm working on freqtrade, a trading cryptocurrency bot library. To alow quicker price analysis I'm adding the possibility to use a maximum of cpu ressources. Here is the issue that I'm working on : issue. And the feature discussion feature The ProblemThe bot run a loop each minute to check price and do analysis. After each loop completion the workers memory rise slowly of about 1%. Then I receive this warning multiple time: At this point the workers start a strike, no more work is performed. Logs, confirming that the workers are restarting due to the client.restart() at each end of loop : What I Tried-start workers on Process mode freqtrade-create-trade-with-dask.zip ResultThe issue is present despite all attempts. During the workers restart there is a deserialisation process from the worker space folder? Do you think that the memory loss can be maintained by that process? |
After a restart workers don't load anything in. They're fresh Python
processes and all previous work is removed and forgotten.
Three thoughts come to mind:
1. Your library code or something else in your environment is large and
taking up space. You should try using fewer workers with more memory.
2. You're generating and destroying tons of small numpy arrays, and
running into a memory leak in Malloc/Numpy:
dask/dask#3530
3. (most likely) something totally different and specific to your
workflow
…On Fri, Aug 31, 2018 at 11:30 AM, Axel CHERUBIN ***@***.***> wrote:
@mrocklin <https://github.com/mrocklin> Thank you for your support.
The context
I'm working on freqtrade <https://github.com/freqtrade/freqtrade>, a
trading cryptocurrency bot library. To alow quicker price analysis I'm
adding the possibility to use a maximum of cpu ressources. Here is the
issue that I'm working on : issue
<freqtrade/freqtrade#746>. And the feature
discussion feature <freqtrade/freqtrade#1170>
The Problem
The bot run a loop each minute to check price and do analysis. After each
loop completion the workers memory rise slowly of about 1%. Then I receive
this warning multiple time:
distributed.utils_perf - WARNING - full garbage collections took 14% CPU
time recently (threshold: 10%)
Then after a few more loops:
distributed.worker - WARNING - Memory use is high but worker has no data
to store to disk. Perhaps some other process is leaking memory? Process
memory: 1.60 GB -- Worker memory limit: 2.09 GB
At this point the workers start a strike, no more work is performed.
Logs, confirming that the workers are restarting due to the
client.restart() at each end of loop :
distributed.scheduler - INFO - Clear task state distributed.scheduler -
INFO - Register inproc://192.168.42.128/8244/2 distributed.scheduler -
INFO - Starting worker compute stream, inproc://192.168.42.128/8244/2
distributed.scheduler - INFO - Register inproc://192.168.42.128/8244/4
distributed.scheduler - INFO - Starting worker compute stream, inproc://
192.168.42.128/8244/4 distributed.scheduler - INFO - Register inproc://
192.168.42.128/8244/6 distributed.scheduler - INFO - Starting worker
compute stream, inproc://192.168.42.128/8244/6 distributed.scheduler -
INFO - Register inproc://192.168.42.128/8244/8 distributed.scheduler -
INFO - Starting worker compute stream, inproc://192.168.42.128/8244/8
distributed.scheduler - INFO - Register inproc://192.168.42.128/8244/10
distributed.scheduler - INFO - Starting worker compute stream, inproc://
192.168.42.128/8244/10 distributed.scheduler - INFO - Register inproc://
192.168.42.128/8244/12 distributed.scheduler - INFO - Starting worker
compute stream, inproc://192.168.42.128/8244/12 distributed.scheduler -
INFO - Register inproc://192.168.42.128/8244/14 distributed.scheduler -
INFO - Starting worker compute stream, inproc://192.168.42.128/8244/14
distributed.scheduler - INFO - Register inproc://192.168.42.128/8244/16
distributed.scheduler - INFO - Starting worker compute stream, inproc://
192.168.42.128/8244/16 distributed.scheduler - INFO - Client
Client-ef7049de-ad17-11e8-a034-5f58bbbb3184 requests to cancel 0 keys
distributed.scheduler - INFO - Send lost future signal to clients
distributed.scheduler - INFO - Remove worker inproc://192.168.42.128/8244/
10 distributed.scheduler - INFO - Remove worker inproc://
192.168.42.128/8244/12 distributed.scheduler - INFO - Remove worker
inproc://192.168.42.128/8244/14 distributed.scheduler - INFO - Remove
worker inproc://192.168.42.128/8244/16 distributed.scheduler - INFO -
Remove worker inproc://192.168.42.128/8244/2 distributed.scheduler - INFO
- Remove worker inproc://192.168.42.128/8244/4 distributed.scheduler -
INFO - Remove worker inproc://192.168.42.128/8244/6 distributed.scheduler
- INFO - Remove worker inproc://192.168.42.128/8244/8
distributed.scheduler - INFO - Lost all workers distributed.scheduler -
INFO - Clear task state distributed.scheduler - INFO - Clear task state
What I Tried
-start workers of Process mode
-start workers on Thread mode
-client.restart()
-Pands option, pd.options.mode.chained_assignment = None
freqtrade-create-trade-with-dask.zip
<https://github.com/dask/distributed/files/2340623/freqtrade-create-trade-with-dask.zip>
Result
The issue is present despite all attempts. During the workers restart
there is a deserialisation process from the worker space folder? Do you
think that the memory loss can be maintained by that process?
Regards,
Axel
—
You are receiving this because you were mentioned.
Reply to this email directly, view it on GitHub
<#2068 (comment)>,
or mute the thread
<https://github.com/notifications/unsubscribe-auth/AASszOImIG3441MpJxqMEyGPIezQ33ZRks5uWValgaJpZM4UzTGE>
.
|
Ok I will try to use fewer workers, just for information i'm running dask distributed 1.23.0. With python 3.7 on a Debian VMware virtual machine. |
@mrocklin as you suggested I tried to run the code with less workers to have more memory for each one. The behaviour is the same, I just have more time before the memory limit is reached. I've noted that there is a mismatch between the used memory by worker showed by Dask dashboard and the total memory used showed by Windows Or Linux. Dask Dashboard = 2Go by worker = 2*16 = 32Go used Do you think that there is a possibility that Dask perform a wrong memory usage calculation then freeze the workers despite there is memory available? |
That number uses psutil.Process().memory_info().rss, which looks at OS
level metrics (assuming you're using a somewhat recent version of Dask).
I'm surprised to hear that they're different.
…On Fri, Aug 31, 2018 at 7:05 PM, Axel CHERUBIN ***@***.***> wrote:
@mrocklin <https://github.com/mrocklin> as you suggested I tried to run
the code with less workers with to have more memory for each one. The
behaviour is the same, I just havec more time before the memory limit is
reached.
I've note that there is a mismath between the used memory by worker showed
by Dask dashboard and the total memory used showed by Windows Or Linux.
Dask Dashboard = 2Go by worker = 2*16 = 32Go used
Windows Ram = 10.4Go used (16Go total)
Do you think that there is a possibility that Dask perform a wrong memory
usage calculation then freeze the workers despite there is memory available?
—
You are receiving this because you were mentioned.
Reply to this email directly, view it on GitHub
<#2068 (comment)>,
or mute the thread
<https://github.com/notifications/unsubscribe-auth/AASszDIt1vWVCG2mJ9Dssx3UWbqtsMZOks5uWcE2gaJpZM4UzTGE>
.
|
I'm using the version 1.23.0 that you pushed yesterday, maybe that function send to dask wrong data? But it does not seem really likely. |
@Axel-CH I've also noticed a mismatch between the memory usage reported by dask distributed and the OS. What helped me to resolve problems of freezed and killed workers was to change the configuration described here to the following:
|
Thanks you for your feedback @abast, I really appreciate it. I will test what you advised, we'll see if it's working in my case. rgds |
Have anyone found out the solution of this problem |
@Athlete369 it's not clear that there's a single problem here, which adds to the confusion. See #2757, dask/dask#3530, among others. |
Interestingly, I get similar results to @mrocklin when doing concurrent POST requests when using partial functions submitted to a threaded local client i.e. client = Client(processes=False)
tasks = [client.submit(functools.partial(requests.post, URL, json=MESSAGE)) for _ in range(MSG_COUNT)]
list(tqdm(as_completed(tasks), total=len(tasks))) leads to
HOWEVER simply passing my arguments to the tasks = [client.submit(requests.post, URL, json=MESSAGE) for _ in range(MSG_COUNT)] |
The issue also goes away if I reduce the size of the payload. Seems like an issue with the size of what's being pickled as suggested by @iljau's investigation |
Any solution or walk-around for this problem? It's really urgent for me now. |
No updates from me. |
sorry , I find it's not memory leak problem in my case. Actually , it seems (personal opinion) the problem of bad control of graph size (I mean number of tasks). If i control the number of tasks submited, the memory is almost at a constant level. Thanks . |
@songqiqqq's suggestion seems to be a viable workaround -- limiting the number of tasks scheduled at any given time. Changing MSG_COUNT = 1_000_000
with Client(processes=False) as client:
pbar = tqdm(total=MSG_COUNT)
tasks = [client.submit(functools.partial(requests.post, URL, json=MESSAGE)) for _ in range(MSG_COUNT)]
for _ in as_completed(tasks):
pbar.update(1) to MSG_COUNT = 1_000_000
def group(iterable, chunk_size=50_000):
iterable = iter(iterable)
chunk = list(it.islice(iterable, chunk_size))
while chunk:
yield chunk
chunk = list(it.islice(iterable, chunk_size))
with Client(processes=False) as client:
pbar = tqdm(total=MSG_COUNT)
for chunk in group(range(MSG_COUNT)):
tasks = [client.submit(functools.partial(requests.post, URL, json=MESSAGE)) for _ in chunk]
for _ in as_completed(tasks):
pbar.update(1) solved the issue for me. I continued to get warnings; but tasks were processed |
FWIW, by clearing def clear_lrus():
import functools
import gc
gc.collect()
wrappers = [a for a in gc.get_objects() if
isinstance(a, functools._lru_cache_wrapper)]
for wrapper in wrappers:
wrapper.cache_clear()
def main_distributed_client():
from distributed import Client
client = Client(processes=True, n_workers=1, threads_per_worker=1)
for i in range(1000):
delayed_obj = Combiner().delayed_results()
future = client.compute(delayed_obj)
res = future.result()
print("#%d, mem: %s mb" % (i, res['mem_usage_mb']))
collect_res = client.run(lambda: gc.collect()) # doesn't help
collect_res = client.run(clear_lrus) In particular, I'm eying on key_split since I'm not certain which kind of objects might end up there. The rest is maybe within worker logs? |
I wanted to add some unexpected results I observed and was able to resolve thanks to the dialogue above... Using I was observing errors while trying to run my Prefect workflows on an AWS Stack involving some ECS Fargate containers, which kept saying something along the lines of this:
This was strange, and I definitely had a few other lingering problems that really made my hunt difficult until I saw some logs that stated the following:
Above, @mrocklin suggested "less workers and more memory per worker", which proved to be my silver bullet in this instance. Hope someone else sees that issue and can react accordingly! |
I have used
dask.delayed
to wire together some classes and when usingdask.threaded.get
everything works properly. When same code is run usingdistributed.Client
memory used by process keeps growing.Dummy code to reproduce issue is below.
Results:
The text was updated successfully, but these errors were encountered: