<commit_before>ARG BASE_IMAGE

FROM nvidia/cuda:${BASE_IMAGE}

RUN apt-get update -y && \
    apt-get install -y --no-install-recommends \
    git \
    cmake \
    libblas3 \
    libblas-dev \
	curl \
	zlib1g-dev \
    libreadline-dev \
    libsqlite3-dev \
    libssl-dev \
	libffi-dev \
	build-essential \
    libbz2-dev \
	ssh \
	wget \
    && \
    rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*

# Install OpenMPI with CUDA
ENV OMPI_VERSION 3.1.3

RUN cd /tmp && wget -q https://www.open-mpi.org/software/ompi/v${OMPI_VERSION%\.*}/downloads/openmpi-$OMPI_VERSION.tar.bz2 && \
  tar -xjf openmpi-$OMPI_VERSION.tar.bz2

RUN cd /tmp/openmpi-$OMPI_VERSION && \
    ./configure --prefix=/usr --with-cuda --disable-oshmem --disable-mpi-java --disable-java --disable-mpi-fortran && \
    make -j 10 && make install && cd && rm -r /tmp/openmpi-$OMPI_VERSION* && \
    /usr/bin/ompi_info --parsable --all | grep -q "mpi_built_with_cuda_support:value:true" && \
    rm -rf /tmp/openmpi-*


# Install pyenv

ENV PYENV_ROOT /usr/local/pyenv
ENV BASH_PROFILE /root/.bash_profile

RUN git clone https://github.com/pyenv/pyenv.git $PYENV_ROOT

RUN touch $BASH_PROFILE && \
    echo 'export PYENV_ROOT="$PYENV_ROOT"' >> $BASH_PROFILE && \
	echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> $BASH_PROFILE && \
	echo 'eval "$(pyenv init -)"' >> $BASH_PROFILE

#
# Install different versions of python and necessary packages
#

<commit_msg>fix comment<commit_after>ARG BASE_IMAGE

FROM nvidia/cuda:${BASE_IMAGE}

RUN apt-get update -y && \
    apt-get install -y --no-install-recommends \
    git \
    cmake \
    libblas3 \
    libblas-dev \
	curl \
	zlib1g-dev \
    libreadline-dev \
    libsqlite3-dev \
    libssl-dev \
	libffi-dev \
	build-essential \
    libbz2-dev \
	ssh \
	wget \
    && \
    rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*

# Install OpenMPI with CUDA
ENV OMPI_VERSION 3.1.3

RUN cd /tmp && wget -q https://www.open-mpi.org/software/ompi/v${OMPI_VERSION%\.*}/downloads/openmpi-$OMPI_VERSION.tar.bz2 && \
  tar -xjf openmpi-$OMPI_VERSION.tar.bz2

RUN cd /tmp/openmpi-$OMPI_VERSION && \
    ./configure --prefix=/usr --with-cuda --disable-oshmem --disable-mpi-java --disable-java --disable-mpi-fortran && \
    make -j 10 && make install && cd && rm -r /tmp/openmpi-$OMPI_VERSION* && \
    /usr/bin/ompi_info --parsable --all | grep -q "mpi_built_with_cuda_support:value:true" && \
    rm -rf /tmp/openmpi-*


# Install pyenv

ENV PYENV_ROOT /usr/local/pyenv
ENV BASH_PROFILE /root/.bash_profile

RUN git clone https://github.com/pyenv/pyenv.git $PYENV_ROOT

RUN touch $BASH_PROFILE && \
    echo 'export PYENV_ROOT="$PYENV_ROOT"' >> $BASH_PROFILE && \
	echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> $BASH_PROFILE && \
	echo 'eval "$(pyenv init -)"' >> $BASH_PROFILE

#
# Install Python and necessary packages
#


{'fraction_non_alphanumeric': 0.15867418899858957, 'fraction_numerical': 0.007757404795486601, 'mean_word_length': 2.757615894039735, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 15, 'https://': 4, 'lorem ipsum': 0, 'www.': 2, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 2, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5089635', 'n_tokens_mistral': 1179, 'n_tokens_neox': 1101, 'n_words': 220}