NVIDIA / gpu-rest-engine

A REST API for Caffe using Docker and Go
BSD 3-Clause "New" or "Revised" License
421 stars 94 forks source link

Will GRE or tensorRT support caffe2? #13

Closed OdinLin closed 7 years ago

OdinLin commented 7 years ago

GRE currently is supporting Caffe, It would be great if support caffe2 framework, any plan?

flx42 commented 7 years ago

@cgottbrath-nvidia can you answer this question?

cgottbrath-nvidia commented 7 years ago

For GRE .. no plans. But it is open source. I would love to see someone adapt it for C2. TensorRT can be used with C2 today using the C++ graph definition API. We are working on an export path for C2 models into TensorRT, so watch for updates in that area.

flx42 commented 7 years ago

Thanks @cgottbrath-nvidia!