Fix pytorch demo CI

This commit is contained in:
Hui, Chunyang 2022-07-11 03:18:22 +00:00 committed by volcano
parent 84b26a8194
commit 016ef88535

@ -271,18 +271,7 @@ jobs:
run: docker exec ${{ github.job }} bash -c "cd /root/occlum/demos/pytorch; ./install_python_with_conda.sh"
- name: Run pytorch test
run: docker exec -d ${{ github.job }} bash -c "cd /root/occlum/demos/pytorch; SGX_MODE=SIM ./run_pytorch_on_occlum.sh 2>&1 | tee /root/occlum/log"
# FIXME: PyTorch can't exit normally in SIM mode
- name: Kill the container
run: |
sleep 360;
cat "$GITHUB_WORKSPACE/log";
if grep -q Done "$GITHUB_WORKSPACE/log"; then
docker stop ${{ github.job }}
else
exit 1
fi
run: docker exec ${{ github.job }} bash -c "cd /root/occlum/demos/pytorch; SGX_MODE=SIM ./run_pytorch_on_occlum.sh"
Tensorflow_test: