From 61103b25eabe4e6c4598d43e5a8bee8232a1896a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?F=C3=A9lix=20Voituret?= Date: Thu, 21 Nov 2019 17:54:25 +0100 Subject: [PATCH] fix: syntx error --- Makefile | 4 ++-- docker/python-3.6-gpu.dockerfile | 9 +++++++++ 2 files changed, 11 insertions(+), 2 deletions(-) create mode 100644 docker/python-3.6-gpu.dockerfile diff --git a/Makefile b/Makefile index 26380a2..609518c 100644 --- a/Makefile +++ b/Makefile @@ -16,12 +16,12 @@ clean: rm -Rf dist build: clean - sed -i "s/project_name = ''/project_name = 'spleeter'/g" + sed -i "s/project_name = ''/project_name = 'spleeter'/g" setup.py sed -i "s/tensorflow_dependency = '[^']*'/tensorflow_dependency = 'tensorflow'/g" setup.py python3 setup.py sdist build-gpu: clean - sed -i "s/project_name = ''/project_name = 'spleeter-gpu'/g" + sed -i "s/project_name = ''/project_name = 'spleeter-gpu'/g" setup.py sed -i "s/tensorflow_dependency = '[^']*'/tensorflow_dependency = 'tensorflow-gpu'/g" setup.py python3 setup.py sdist diff --git a/docker/python-3.6-gpu.dockerfile b/docker/python-3.6-gpu.dockerfile new file mode 100644 index 0000000..d42f3db --- /dev/null +++ b/docker/python-3.6-gpu.dockerfile @@ -0,0 +1,9 @@ +FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 + +# TODO: Install cuda ! +RUN apt-get update && apt-get install -y ffmpeg libsndfile1 +RUN pip install musdb museval +RUN pip install spleeter-gpu==1.4.4 +RUN mkdir -p /model +ENV MODEL_PATH /model +ENTRYPOINT ["spleeter"] \ No newline at end of file