[sound-spleeter] Using GPU.

singnet · May 15, 2020 · e365f1c · e365f1c
1 parent b750296
commit e365f1c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/services/sound-spleeter/Dockerfile b/services/sound-spleeter/Dockerfile
@@ -1,4 +1,4 @@
-FROM ubuntu:18.04
+FROM tensorflow/tensorflow:1.15.2-gpu-py3
 
 ARG git_owner="singnet"
 ARG git_repo="dnn-model-services"

diff --git a/services/sound-spleeter/requirements.txt b/services/sound-spleeter/requirements.txt
@@ -1,5 +1,5 @@
 grpcio>=1.14.2
 grpcio-tools>=1.14.1
 requests==2.20.0
-spleeter
+spleeter-gpu
 pydub==0.23.1