chore: adding wget

Author: Hartorn

build-dockerfiles.sh

@@ -50,8 +50,12 @@ for python_version in "3.7" "3.8" "3.9-rc"; do
         if [ "$type" == "gpu" ]; then
             echo "" >>"${output_file}"
             echo "# Adding Nvidia ML repo" >>"${output_file}"
+            
             echo 'RUN echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list' >>"${output_file}"
-            echo 'RUN wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | sudo apt-key add -' >>"${output_file}"
+            apt_install_temp_packages ${output_file} "wget"
+            echo '&& wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - \' >>"${output_file}"
+            apt_clean_temp_packages ${output_file}
+            echo "" >>"${output_file}"
 
             echo "" >>"${output_file}"
             echo "Adding CUDNN"

dockerfiles/3.7/gpu/Dockerfile

@@ -191,7 +191,17 @@ RUN set -ex; \
 
 # Adding Nvidia ML repo
 RUN echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
-RUN wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | sudo apt-key add -
+RUN savedAptMark="$(apt-mark showmanual)" \
+&& apt update \
+&& apt install wget -y --no-install-recommends \
+&& wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - \
+&& apt-mark auto '.*' > /dev/null \
+&& apt-mark manual $savedAptMark \
+&& apt purge -y --auto-remove -o APT::AutoRemove::RecommendsImportant=false \
+&& apt autoclean \
+&& apt clean \
+&& rm -rf /var/lib/apt/lists/*
+
 
 # Adding CUDNN
 ENV CUDNN_VERSION 8.0.2.39

dockerfiles/3.8/gpu/Dockerfile

@@ -155,7 +155,17 @@ RUN set -ex; \
 
 # Adding Nvidia ML repo
 RUN echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
-RUN wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | sudo apt-key add -
+RUN savedAptMark="$(apt-mark showmanual)" \
+&& apt update \
+&& apt install wget -y --no-install-recommends \
+&& wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - \
+&& apt-mark auto '.*' > /dev/null \
+&& apt-mark manual $savedAptMark \
+&& apt purge -y --auto-remove -o APT::AutoRemove::RecommendsImportant=false \
+&& apt autoclean \
+&& apt clean \
+&& rm -rf /var/lib/apt/lists/*
+
 
 # Adding CUDNN
 ENV CUDNN_VERSION 8.0.2.39

dockerfiles/3.9-rc/gpu/Dockerfile

@@ -154,7 +154,17 @@ RUN set -ex; \
 
 # Adding Nvidia ML repo
 RUN echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
-RUN wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | sudo apt-key add -
+RUN savedAptMark="$(apt-mark showmanual)" \
+&& apt update \
+&& apt install wget -y --no-install-recommends \
+&& wget -qO - https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - \
+&& apt-mark auto '.*' > /dev/null \
+&& apt-mark manual $savedAptMark \
+&& apt purge -y --auto-remove -o APT::AutoRemove::RecommendsImportant=false \
+&& apt autoclean \
+&& apt clean \
+&& rm -rf /var/lib/apt/lists/*
+
 
 # Adding CUDNN
 ENV CUDNN_VERSION 8.0.2.39