diff --git a/Dockerfile b/Dockerfile
index 25a002d..f4662f6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,4 +5,10 @@ ARG NB_UID="1000"
 ARG NB_GID="100"
 ENV CONDA_CUDA_OVERRIDE="12.1"  
 ENV NUM_GPUS=1
+USER root
+RUN apt-get install --yes --no-install-recommends \
+    ffmpeg  \
+    libportaudio2  \
+    apt-get clean && rm -rf /var/lib/apt/lists/* 
+USER $NB_UID
 RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
\ No newline at end of file