Refer the below thread, it might help. Although it’s specific to colab, you might find workarounds for your environment -