Added caffe2 tests to pipeline, caffe2 docker image

parent f6a97730
Pipeline #109857 failed with stage
in 5 minutes and 7 seconds
......@@ -40,6 +40,14 @@ integrationMXNetJobLinux:
- mvn -Dorg.slf4j.simpleLogger.log.org.apache.maven.cli.transfer.Slf4jMavenTransferListener=warn -B clean install --settings settings.xml -Dtest=IntegrationMXNetTest
only:
- oneclick_nn_training
integrationCaffe2JobLinux:
stage: linux
image: registry.git.rwth-aachen.de/monticore/embeddedmontiarc/generators/emadl2cpp/integrationtests/caffe2:v0.0.1
script:
- mvn -Dorg.slf4j.simpleLogger.log.org.apache.maven.cli.transfer.Slf4jMavenTransferListener=warn -B clean install --settings settings.xml -Dtest=IntegrationCaffe2Test
only:
- oneclick_nn_training
masterJobWindows:
stage: windows
......
FROM maven:3-jdk-8
RUN apt-get update
RUN apt-get install -y --no-install-recommends \
build-essential \
cmake \
git \
libgoogle-glog-dev \
libgtest-dev \
libiomp-dev \
libleveldb-dev \
liblmdb-dev \
libopencv-dev \
libopenmpi-dev \
libsnappy-dev \
libprotobuf-dev \
openmpi-bin \
openmpi-doc \
protobuf-compiler \
python-dev \
python-pip
RUN apt-get install -y --no-install-recommends python-setuptools
RUN pip install \
future \
numpy \
protobuf
RUN apt-get install -y --no-install-recommends libgflags-dev
ENV NAME World
ENV PYTHONPATH="/usr/local"
ENV PYTHONPATH="${PYTHONPATH}:/pytorch/build"
ENV PYTHONPATH="${PYTHONPATH}:/usr/bin/python"
ENV LD_LIBRARY_PATH="/usr/local/lib"
RUN git clone --recursive https://github.com/pytorch/pytorch.git && cd pytorch
RUN cd pytorch && git checkout v0.4.0
RUN cd pytorch && git submodule update --init
RUN cd pytorch && mkdir build && cd build
RUN cd pytorch && cd build && cmake -DUSE_MPI=OFF ..
RUN cd pytorch && cd build && make install
RUN python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment