config_gpu.pbtxt 277 B

1234567891011121314151617181920212223
  1. backend: "python"
  2. max_batch_size: 1
  3. input [
  4. {
  5. name: "input"
  6. data_type: TYPE_STRING
  7. dims: [ 1 ]
  8. }
  9. ]
  10. output [
  11. {
  12. name: "output"
  13. data_type: TYPE_STRING
  14. dims: [ 1 ]
  15. }
  16. ]
  17. instance_group [
  18. {
  19. count: 1
  20. kind: KIND_GPU
  21. gpus: [ 0 ]
  22. }
  23. ]