tf_gpu_manager
tf_gpu_manager copied to clipboard
called by multiprocessing[suggestion]
Hi, I am trying to use the code for a multiprocessing job.
The major task is that there are several processes and some of them might call the gpu_manager at a time. The problem is the shared unspecified_gpu (dict) need to be locked when calling by one process. I suggest adding a lock in your code to make sure it is working in multiprocessing task.
Actually, the lock should be shared by gpu_manager and the other processes. It can be claimed as global variable in init().
Thank you for this issue,i will try to fix it.