Update Dockerfile pytorch_lightning

ucsd-ets · May 28, 2024 · cf00800 · cf00800
1 parent 1e1f8b1
commit cf00800
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -22,7 +22,7 @@ USER jovyan
 
 # RUN conda install -y scikit-learn
 
-RUN pip install --no-cache-dir networkx scipy wave soundfile
+RUN pip install --no-cache-dir networkx scipy wave soundfile pytorch_lightning
 
 # Override command to disable running jupyter notebook at launch
 # CMD ["/bin/bash"]