diff --git a/base-devel.dockerfile b/base-devel.dockerfile
index 3776a3b..a04552e 100644
--- a/base-devel.dockerfile
+++ b/base-devel.dockerfile
@@ -21,6 +21,8 @@ RUN ln -s /usr/bin/python3 /usr/bin/python
 RUN pip install \
     numpy \
     torch \
+    torchvision \
+    torchaudio  \
     jupyterlab
 
 # start jupyter lab
diff --git a/base.dockerfile b/base.dockerfile
index d1ad3bb..ebe8473 100644
--- a/base.dockerfile
+++ b/base.dockerfile
@@ -20,6 +20,8 @@ RUN ln -s /usr/bin/python3 /usr/bin/python
 RUN pip install \
     numpy \
     torch \
+    torchvision \
+    torchaudio  \
     jupyterlab
 
 # start jupyter lab