Dockerfile_tensorflow

#==================================================================================
#       Copyright (c) 2023 Northeastern University
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#          http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#==================================================================================

FROM nvidia/cuda:12.2.0-runtime-ubuntu20.04
ARG password=ChangeMe

# Set timezone
ENV TZ=America/New_York
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
RUN apt-get update && apt-get install -y openssh-server htop nano vim

# These are required to enable ssh connection to use it on Colosseum
RUN ssh-keygen -A
RUN mkdir -p /run/sshd
RUN echo "root:${password}" | chpasswd
RUN sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/g' /etc/ssh/sshd_config

# Install system dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
    git \
    curl \
    wget \
    build-essential \
    cmake \
    libopenblas-dev \
    liblapack-dev \
    libhdf5-dev \
    libopencv-dev \
    libssl-dev \
    libffi-dev \
    libpng-dev \
    libjpeg-dev \
    libtiff-dev \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender-dev \
    python3-dev \
    python3-pip \
    python3-setuptools \
    python3-wheel \
    && rm -rf /var/lib/apt/lists/*

# Install TensorFlow with GPU support
RUN pip3 install --no-cache-dir tensorflow-gpu==2.7.0
RUN pip3 install protobuf==3.20.*

# Set up working directory
WORKDIR /workspace

# This command will be always overwritten by the system
CMD ["/usr/sbin/sshd", "-D"]