Question
Tensorflow, torch memory not enough error.
- Posted on July 31, 2023
- Python Modules
Asked by 0cce417fb7c54f349df192251f2877
As following some tutorials on web app with flask. I made a trained model flask app with Tensorflow and torch, however I have tried many times and different lite version of them both they will not get install. The code I have is here in the GitHub. I have trained model in pickle stored in spaces. However I have not been able to deploy with even large app RAM.
Thank in advance for the help.
This textbox defaults to using Markdown to format your answer.
You can type !ref in this text area to quickly search our full set of tutorials, documentation & marketplace offerings and insert the link!
These answers are provided by our Community. If you find them useful, show some love by clicking the heart. If you run into issues leave a comment, or add your own answer to help others.
Hollie's Hub for Good
Working on improving health and education, reducing inequality, and spurring economic growth? We’d like to help.
Get started for free
Enter your email to get $200 in credit for your first 60 days with DigitalOcean.
New accounts only. By submitting your email you agree to our Privacy Policy.
Hi there,
Indeed, apps that use libraries like TensorFlow and PyTorch have high memory requirements.
What I could suggest is:
If this still does not help, there are a few other things that you could try like:
Find lightweight versions or alternatives to these libraries. For example, consider using TensorFlow Lite or PyTorch Mobile, which are designed to be used on devices with less computational power.
You can try reducing the size of your model. Here are some tips on how to do that:
Alternatively, consider using a different framework like Streamlit that is built to handle ML and DL models in a memory-efficient way.
Hope that this helps!
Best,
Bobby