search

UMass-Rescue / CombinedTechStack

Handle and process large amounts of media data with plug-and-play machine learning models
https://umass-rescue.github.io/CombinedTechStack/
MIT License
1 stars 1 forks source link

Enable GPU In Microservices #11

Open CodyRichter opened 3 years ago

CodyRichter commented 3 years ago

Currently, the Docker containers used for training and prediction microservices only have access to the CPU and memory. Allowing the containers to interact with the system GPU can lead to large speed-ups in prediction and training tasks.