alihan/python-gpu
1
0
Fork 0
mirror of https://github.com/superlinear-ai/python-gpu.git synced 2025-03-26 03:37:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: optimize Docker image size (#6)

Browse Source
This commit is contained in:
Laurent Sorber
2023-09-11 08:58:08 +02:00
committed by GitHub
parent 6f3b8c35ac
commit 231e36e330
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile
View File

@@ -1,6 +1,6 @@
# syntax=docker/dockerfile:1
ARG PYTHON_VERSION=3.8
FROM python:$PYTHON_VERSION-slim AS base
FROM python:$PYTHON_VERSION-slim
# Install wget.
RUN apt-get update && \
@@ -11,7 +11,8 @@ RUN apt-get update && \
# Install Miniconda.
RUN CONDA_ARCH=$([ "$TARGETARCH" = "arm64" ] && echo "aarch64" || echo "x86_64") && \
wget --quiet "https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-$CONDA_ARCH.sh" --output-document ~/miniconda.sh && \
/bin/bash ~/miniconda.sh -b -p /opt/conda
/bin/bash ~/miniconda.sh -b -p /opt/conda && \
rm ~/miniconda.sh
ENV PATH=/opt/conda/bin:$PATH
ENV LD_LIBRARY_PATH=/opt/conda/lib:$LD_LIBRARY_PATH
@@ -20,4 +21,5 @@ ARG CUDA_VERSION=11.8
ARG CUDNN_VERSION=8.8
RUN conda install --name base conda-libmamba-solver && \
conda config --set solver libmamba && \
conda install --channel conda-forge --yes cudatoolkit="$CUDA_VERSION" cudnn="$CUDNN_VERSION"
conda install --name base --channel conda-forge --yes cudatoolkit="$CUDA_VERSION" cudnn="$CUDNN_VERSION" && \
conda clean --all --force-pkgs-dirs --yes