install torchvision and torchaudio for base images

Tverous · web-flow · commit d6245fc4109b · 2024-04-06T21:12:19.000-04:00
diff --git 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
@@ -20,6 +20,8 @@ RUN ln -s /usr/bin/python3 /usr/bin/python
 RUN pip install \
     numpy \
     torch \
+    torchvision \
+    torchaudio  \
     jupyterlab
 
 # start jupyter lab
