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