RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock. self.pid = os.fork()

[melikaaala]

Hi
I have trained machine learning python code on google colab but I faced this error. Is there any solution for this:
os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock. self.pid = os.fork()

[Delgan]

Can you please share the code that produces this warning?

Loguru does not call os.fork().

[melikaaala]

I use this url: https://github.com/Algolzw/image-restoration-sde

[melikaaala]

hi I use this code: https://github.com/Algolzw/image-restoration-sde

…

On Tue, Apr 16, 2024 at 9:29 AM Delgan ***@***.***> wrote: Can you please share the code that produces this warning? Loguru does not call os.fork(). — Reply to this email directly, view it on GitHub <#1125 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ALIF3BCCZTOHLUSXTG3NVR3Y5S43HAVCNFSM6AAAAABGISM4LSVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDANJYGI4DQMRYGY> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[Delgan]

Sorry but there is not enough information for me to help you. Please, try to provide a minimal reproducible example involving usage of Loguru.

Right now, I don't understand how the problem you're facing is related to Loguru. As I said, your error is complaining about usage of os.fork() within JAX framework, but Loguru does not make any use of forking.

[steveepreston]

this bug still exists in tf 2.16.1

[MAJK1988]

Ensure that you have the latest version of JAX installed. There may be fixes in more recent versions that address these issues. You can upgrade JAX and its dependencies as follows:

pip install --upgrade jax jaxlib