diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index afb7d6dd8c427c92a28d12e1bb8985942d012a62..05c396baa2e5ef8e0754ee88ff1e1fd2d5a0b27c 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -1,6 +1,11 @@
 name: docker
 on:
-  - workflow_dispatch
+  workflow_dispatch:
+    inputs:
+      version:
+        description: "Spleeter version to build image for"
+        required: true
+        default: "2.1.2"
 jobs:
   build-test-push:
     runs-on: ubuntu-latest
@@ -19,7 +24,6 @@ jobs:
           echo "::set-env name=base::python:${{ matrix.distribution }}"
           echo "::set-env name=tag::${{ matrix.distribution }}"
           echo "::set-env name=file::spleeter"
-          echo "::set-env name=package::spleeter"
       - if: ${{ matrix.distribution == 'conda' }}
         name: Build Conda base image
         run: |
@@ -31,17 +35,16 @@ jobs:
           docker build \
             --build-arg BASE=python:${{ matrix.distribution }} \
             -t cuda:${{ matrix.distribution }} \
-            -f docker/cuda-10-0.dockerfile .
+            -f docker/cuda-10-1.dockerfile .
           echo "::set-env name=base::cuda:${{ matrix.distribution }}"
           echo "::set-env name=tag::${{ matrix.distribution }}-gpu"
-          echo "::set-env name=package::spleeter-gpu"
       # ----------------------------------------------------------------------
       # Note: image building.
       - name: Build deezer/spleeter:${{ env.tag }} image
         run: |
           docker build \
             --build-arg BASE=${{ env.base }} \
-            --build-arg SPLEETER_PACKAGE=${{ env.package }} \
+            --build-arg SPLEETER_VERSION=${{ github.event.inputs.version }} \
             -t deezer/spleeter:${{ env.tag }} \
             -f docker/${{ env.file }}.dockerfile .
           echo "::set-env name=modelargs::"
@@ -54,7 +57,7 @@ jobs:
             -t deezer/spleeter:${{ env.tag }}-${{ matrix.model }} \
             -f docker/spleeter-model.dockerfile .
           echo "::set-env name=tag::${{ env.tag }}-${{ matrix.model }}"
-          echo "::set-env name=modelarg::-p spleeter:${{ matrix.model }}"          
+          echo "::set-env name=modelarg::-p spleeter:${{ matrix.model }}"
       # ----------------------------------------------------------------------
       # Note: image testing.
       - name: Test deezer/spleeter:${{ env.tag }} image
diff --git a/docker/spleeter-conda.dockerfile b/docker/spleeter-conda.dockerfile
index 2a21cd86b7d0ec6bd6905b4cf37dac3405955737..4a8f704b9623a6c5ab3adfc30a20776acc56996b 100644
--- a/docker/spleeter-conda.dockerfile
+++ b/docker/spleeter-conda.dockerfile
@@ -2,12 +2,12 @@ ARG BASE=conda
 
 FROM ${BASE}
 
-ARG SPLEETER_PACKAGE=spleeter
 ARG SPLEETER_VERSION=1.5.3
 ENV MODEL_PATH /model
 
 RUN mkdir -p /model
+RUN conda config --add channels conda-forge
 RUN conda install -y -c conda-forge musdb
-RUN conda install -y -c conda-forge ${SPLEETER_PACKAGE}==${SPLEETER_VERSION}
+RUN conda install -y -c deezer-research spleeter 
 COPY docker/conda-entrypoint.sh spleeter-entrypoint.sh
 ENTRYPOINT ["/bin/bash", "spleeter-entrypoint.sh"]
\ No newline at end of file
diff --git a/docker/spleeter.dockerfile b/docker/spleeter.dockerfile
index 1161c3aad65851643dd738479bbc7cdd6e640312..745ceae7388e1d95088bf446b63250fed58af949 100644
--- a/docker/spleeter.dockerfile
+++ b/docker/spleeter.dockerfile
@@ -2,13 +2,12 @@ ARG BASE=python:3.6
 
 FROM ${BASE}
 
-ARG SPLEETER_PACKAGE=spleeter
 ARG SPLEETER_VERSION=1.5.3
 ENV MODEL_PATH /model
 
 RUN mkdir -p /model
 RUN apt-get update && apt-get install -y ffmpeg libsndfile1
 RUN pip install musdb museval
-RUN pip install ${SPLEETER_PACKAGE}==${SPLEETER_VERSION}
+RUN pip install spleeter==${SPLEETER_VERSION}
 
 ENTRYPOINT ["spleeter"]