fix: syntx error

This commit is contained in:
Félix Voituret
2019-11-21 17:54:25 +01:00
parent d5bdb7ee66
commit 61103b25ea
2 changed files with 11 additions and 2 deletions

View File

@@ -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

View File

@@ -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"]