diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index 31adda00e64a26667f529f245519f66115433550..d8196145c767881d0ec3796084fb4329715b5275 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -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 diff --git a/src/test/resources/docker/caffe2/Dockerfile b/src/test/resources/docker/caffe2/Dockerfile new file mode 100644 index 0000000000000000000000000000000000000000..a969816c7f41ec3e2b241ed467413530e5ed656d --- /dev/null +++ b/src/test/resources/docker/caffe2/Dockerfile @@ -0,0 +1,44 @@ +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" +