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

TensorRT bumped to 2.1.2.x86_64.cuda-8.0-16-04 #14

Closed drewfarris closed 7 years ago

drewfarris commented 7 years ago

Incorporates TensorRT 2.1.2 & includes minor fix to work around missing yaml in /caffe/models/bvlc_alexnet/readme.md in https://github.com/NVIDIA/caffe/tree/caffe-0.16

Tested on EC2 Centos 7.3.1611 with NVIDIA drivers 375.66, docker-ce-17.03.1, nvidia-docker-1.0.1-1

flx42 commented 7 years ago

Pushed, thanks!