From 6136033b3e13f1721d58a5515633304079ddae16 Mon Sep 17 00:00:00 2001 From: Michelangelo Dagostino Date: Thu, 2 Nov 2023 12:05:40 -0500 Subject: [PATCH] Two more optimizations. --- ubuntu/gpu/cuda-11.8/venv/Dockerfile | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/ubuntu/gpu/cuda-11.8/venv/Dockerfile b/ubuntu/gpu/cuda-11.8/venv/Dockerfile index 3921049..00a552b 100644 --- a/ubuntu/gpu/cuda-11.8/venv/Dockerfile +++ b/ubuntu/gpu/cuda-11.8/venv/Dockerfile @@ -10,6 +10,8 @@ RUN apt-get update \ && curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \ && /usr/bin/python3.10 get-pip.py pip==22.3.1 setuptools==65.6.3 wheel==0.38.4 \ && rm get-pip.py + && apt-get clean \ + && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* # virtualenv 20.0.24 introduced a periodic update feature, which attempts to update all @@ -46,13 +48,10 @@ RUN /databricks/python3/bin/pip install \ protobuf==4.23.3 \ grpcio==1.48.2 \ grpcio-status==1.48.1 \ - databricks-sdk==0.1.6 + databricks-sdk==0.1.6 \ + && /databricks/python3/bin/pip cache purge # Specifies where Spark will look for the python binary ENV PYSPARK_PYTHON=/databricks/python3/bin/python3 - -# Use pip cache purge to cleanup the cache safely -RUN /databricks/python3/bin/pip cache purge -