Added dependencies in Docker (#141)

TorchDSP · Jun 19, 2023 · 8ee508a · 8ee508a
1 parent 3afde59
commit 8ee508a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -2,13 +2,17 @@ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 ENV DEBIAN_FRONTEND=noninteractive
 
+RUN apt-get update && apt-get install -y \
+    libgl1-mesa-glx && \
+    rm -rf /var/lib/apt/lists/*
+
 ADD torchsig/ /build/torchsig
 
 ADD pyproject.toml /build/pyproject.toml
 
 RUN pip3 install /build
 
-RUN pip3 install notebook
+RUN pip3 install notebook jupyterlab
 
 WORKDIR /workspace/code