/docker-flask-gpu

A Flask API running on a GPU cluster with Docker Swarm

Primary LanguagePython

Docker Flask API on GPU

Installation

Requirements :

  • CUDA drivers
  • Docker
  • nvidia-docker