diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile index f12bb9bc..3f650f64 100644 --- a/images/scipy-ml-notebook/Dockerfile +++ b/images/scipy-ml-notebook/Dockerfile @@ -91,6 +91,8 @@ 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 && pip cache purge + USER $NB_UID:$NB_GID ENV PATH=${PATH}:/usr/local/nvidia/bin:/opt/conda/bin