install torchvision and torchaudio for base images

Tverous · Apr 7, 2024 · d6245fc · d6245fc
1 parent 4d8f63e
commit d6245fc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
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