Skip to content

Commit

Permalink
Update Dockerfile to simplify install
Browse files Browse the repository at this point in the history
  • Loading branch information
satra authored Apr 17, 2024
1 parent 22b34b6 commit 63566dd
Showing 1 changed file with 2 additions and 13 deletions.
15 changes: 2 additions & 13 deletions voice/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,23 +1,12 @@
# use base image
FROM quay.io/ohsu-comp-bio/bridge2ai-jupyter:restricted-gpu-root
FROM quay.io/ohsu-comp-bio/bridge2ai-jupyter:cpu-root

# return to non-root user
USER $NB_UID
WORKDIR /home/$NB_USER

# create environment
# setup environment
RUN conda create -n voice
SHELL ["conda", "run", "-n", "voice", "/bin/bash", "-c"]
RUN conda install python=3.11
RUN pip install --no-cache-dir b2aiprep ipympl huggingface_hub[cli]
RUN pip install ipykernel
RUN python -m ipykernel install --user --name voice --display-name "Python (voice)"

# download models
RUN MODEL_DIR=$HOME/models && \
huggingface-cli download --local-dir $MODEL_DIR/speechbrain/spkrec-ecapa-voxceleb speechbrain/spkrec-ecapa-voxceleb --cache-dir ${MODEL_DIR}/cache && \
huggingface-cli download --local-dir $MODEL_DIR/openai/whisper-base openai/whisper-base --cache-dir ${MODEL_DIR}/cache

# allow use of conda envs
RUN conda init bash && exec bash
RUN conda init bash && exec bash

0 comments on commit 63566dd

Please sign in to comment.