Update tensorflow demos with copy_bom

This commit is contained in:
jianfengjiang 2021-09-26 11:46:38 +08:00 committed by Zongmin.Gu
parent 3f78b09df7
commit 436e427fbf
4 changed files with 43 additions and 20 deletions

@ -27,18 +27,8 @@ new_json="$(jq '.resource_limits.user_space_size = "7000MB" |
echo "${new_json}" > Occlum.json
# 2. Copy files into Occlum Workspace and Build
mkdir -p image/model
cp -rf ../resnet50-v15-fp32 image/model/
cp -rf ../ssl_configure image/bin/
cp ../tensorflow_model_server image/bin
cp ../hosts image/etc/
cp $occlum_glibc/libdl.so.2 image/$occlum_glibc
cp $occlum_glibc/librt.so.1 image/$occlum_glibc
cp $occlum_glibc/libm.so.6 image/$occlum_glibc
cp $occlum_glibc/libutil.so.1 image/$occlum_glibc
cp $occlum_glibc/libpthread.so.0 image/$occlum_glibc
cp $occlum_glibc/libnss_files.so.2 image/$occlum_glibc
cp $occlum_glibc/libnss_compat.so.2 image/$occlum_glibc
rm -rf image
copy_bom -f ../tensorflow_serving.yaml --root image --include-dir /opt/occlum/etc/template
#occlum build
occlum build

@ -0,0 +1,20 @@
includes:
- base.yaml
targets:
# copy model
- target: /model
copy:
- dirs:
- ../resnet50-v15-fp32
- target: /bin
copy:
- from: ..
dirs:
- ssl_configure
files:
- tensorflow_model_server
# copy hosts
- target: /etc
copy:
- files:
- ../hosts

@ -17,14 +17,8 @@ fi
cd occlum_instance
# Copy files into Occlum Workspace and build
if [ ! -L "image/bin/python3" ];then
mkdir -p image/opt
cp -rf $python_dir image/opt/python-occlum
ln -s /opt/python-occlum/bin/python3 image/bin/python3
cp -f /opt/occlum/glibc/lib/libdl.so.2 image/opt/occlum/glibc/lib/
cp -f /opt/occlum/glibc/lib/libutil.so.1 image/opt/occlum/glibc/lib/
cp -f /opt/occlum/glibc/lib/librt.so.1 image/opt/occlum/glibc/lib/
cp -f ../demo.py image/bin
cp -f ../mnist.npz image/bin
rm -rf image
copy_bom -f ../tensorflow_training.yaml --root image --include-dir /opt/occlum/etc/template
new_json="$(jq '.resource_limits.user_space_size = "5400MB" |
.resource_limits.kernel_space_heap_size = "512MB" |
.process.default_mmap_size = "5000MB" |

@ -0,0 +1,19 @@
includes:
- base.yaml
targets:
- target: /bin
createlinks:
- src: /opt/python-occlum/bin/python3
linkname: python3
# python packages
- target: /opt
copy:
- dirs:
- ../python-occlum
# below are python code and data
- target: /bin
copy:
- from: ..
files:
- demo.py
- mnist.npz