Skip to content

Commit 37a1464

Browse files
authored
Update Dockerfile
1 parent 102bf56 commit 37a1464

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

images/scipy-ml-notebook/Dockerfile

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,9 @@ RUN pip install nvidia-cudnn-cu12==$CUDNN_VERSION torch==$TORCH_VERSION torchvis
9595
mamba clean -a -y && \
9696
pip cache purge
9797

98+
#upgrade scikit-learn
99+
RUN pip install --no-cache-dir --upgrade scikit-learn
100+
98101
RUN pip install transformers datasets accelerate huggingface-hub timm && \
99102
fix-permissions $CONDA_DIR && \
100103
fix-permissions /home/$NB_USER && \

0 commit comments

Comments
 (0)