Update Dockerfile (#87)

ucsd-ets · Apr 1, 2024 · a38ee32 · a38ee32
1 parent db59229
commit a38ee32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
@@ -94,7 +94,7 @@ RUN pip install torch==2.1.2 torchvision torchaudio --index-url https://download
   rm /opt/conda/lib/python3.9/site-packages/torch/lib/libcudnn_ops_train.so.8 && \
   rm /opt/conda/lib/python3.9/site-packages/torch/lib/libcublas.so.11
 
-RUN pip install transformers datasets accelerate huggingface-cli && pip cache purge
+RUN pip install transformers datasets accelerate huggingface-cli timm && pip cache purge
 
 USER $NB_UID:$NB_GID
 ENV PATH=${PATH}:/usr/local/nvidia/bin:/opt/conda/bin