torch-Dockerfile

FROM  nvcr.io/nvidia/pytorch:19.07-py3
LABEL maintainer caffe-maint@googlegroups.com

RUN DEBIAN_FRONTEND=noninteractive apt-get update && apt-get install -y --no-install-recommends \
        aria2 \
        git \
        wget \
        zsh \
        vim \
        sudo \
        zsh \
        htop \
        tree \
        tmux \
        rsync \
        imagemagick \
        fdupes \
        openssl \
        libcurl4-gnutls-dev \
        wget \
        openssh-server \
        curl &&\
        rm -rf /var/lib/apt/lists/* &&\
        apt-get clean

RUN gpg --keyserver ha.pool.sks-keyservers.net --recv-keys B42F6819007F00F88E364FD4036A9C25BF357DD4 && \
    curl -o /usr/local/bin/gosu -SL "https://github.com/tianon/gosu/releases/download/1.10/gosu-$(dpkg --print-architecture)" \
        && curl -o /usr/local/bin/gosu.asc -SL "https://github.com/tianon/gosu/releases/download/1.10/gosu-$(dpkg --print-architecture).asc" \
            && gpg --verify /usr/local/bin/gosu.asc \
                && rm /usr/local/bin/gosu.asc \
                    && chmod +x /usr/local/bin/gosu

#RUN apt-get update
#RUN apt-get install -y software-properties-common
#RUN add-apt-repository -y ppa:george-edison55/cmake-3.x
#RUN apt-get install -y cmake
 

# Pip installs

#ld config over ssh
RUN mkdir /var/run/sshd
RUN ldconfig

# Rmate
RUN curl -Lo /bin/rmate https://raw.githubusercontent.com/textmate/rmate/master/bin/rmate && \
        chmod a+x /bin/rmate

# Oh my zsh
RUN ["apt-get", "update"]
RUN ["apt-get", "install", "-y", "zsh"]
RUN wget https://github.com/robbyrussell/oh-my-zsh/raw/master/tools/install.sh -O - | zsh || true

RUN sed -i 's/PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
RUN  yes y | ssh-keygen -P "" -t dsa -f /etc/ssh/ssh_host_dsa_key > /dev/null

COPY entrypoint.sh /usr/local/bin/entrypoint.sh
ENTRYPOINT ["sh","/usr/local/bin/entrypoint.sh"]

WORKDIR /workspace