freeze_requirements.sh 453 B

12345678910111213141516
  1. #!/usr/bin/env bash
  2. # Can we just do this in the Dockerfile? Like, use cache mounts?
  3. for device_type in 'gpu' 'cpu'; do
  4. docker run \
  5. -it \
  6. -e DEVICE_TYPE="${device_type}" \
  7. -e OUID="$(id -u)" \
  8. -e OGID="$(id -g)" \
  9. -v "$(pwd)/../../..":/workspace \
  10. -w /workspace/deploy/hps/server_env \
  11. --rm \
  12. "paddlex-hps-rc:${device_type}" \
  13. /bin/bash scripts/_freeze_requirements.sh
  14. done