2024-11-07 13:17:29 -05:00
|
|
|
#!/bin/bash
|
|
|
|
|
|
2024-09-07 23:48:40 +05:30
|
|
|
# This script build the CPU docker image and run the offline inference inside the container.
|
|
|
|
|
# It serves a sanity check for compilation and basic model usage.
|
|
|
|
|
set -ex
|
|
|
|
|
|
|
|
|
|
# Setup cleanup
|
2024-11-23 15:03:53 +05:30
|
|
|
remove_docker_container() { docker rm -f cpu-test || true; docker system prune -f; }
|
2024-09-07 23:48:40 +05:30
|
|
|
trap remove_docker_container EXIT
|
|
|
|
|
remove_docker_container
|
|
|
|
|
|
2024-11-23 15:03:53 +05:30
|
|
|
# Try building the docker image
|
|
|
|
|
docker build -t cpu-test -f Dockerfile.ppc64le .
|
2024-11-09 11:27:11 +08:00
|
|
|
|
