ARM support #24

Author: rafie

Dockerfile

@@ -1,14 +1,12 @@
 # BUILD redisai-cpu-${ARCH}-${OSNICK}:M.m.b
 
-# stretch|bionic|buster
-ARG OSNICK=buster
+# OSNICK=bionic|stretch|buster
+ARG OSNICK=stretch
 
 #----------------------------------------------------------------------------------------------
 # FROM redisfab/redis-x64-${OSNICK}:5.0.5 AS builder
 FROM redis:latest AS builder
 
-ENV X_NPROC "cat /proc/cpuinfo|grep processor|wc -l"
-
 ADD ./ /build
 WORKDIR /build
 

Dockerfile.arm

@@ -23,6 +23,8 @@ FROM redisfab/redis-${OSNICK}-xbuild:5.0.5
 
 RUN [ "cross-build-start" ]
 
+RUN set -e; apt-get -qq update; apt-get -q install -y libgomp1
+
 ENV LD_LIBRARY_PATH /usr/lib/redis/modules/
 
 RUN mkdir -p "$LD_LIBRARY_PATH"

README.md

@@ -27,7 +27,7 @@ docker run -p 6379:6379 -it --rm redisai/redisai
 
 On the client, load the model
 ```sh
-redis-cli -x AI.MODELSET foo TF CPU INPUTS a b OUTPUTS c < examples/models/graph.pb
+redis-cli -x AI.MODELSET foo TF CPU INPUTS a b OUTPUTS c < test/test_data/graph.pb
 ```
 
 Then create the input tensors, run the computation graph and get the output tensor (see `load_model.sh`). Note the signatures: