From b529cc9e2dda9bd431aaa1d51b5f4e4d1ac75ebd Mon Sep 17 00:00:00 2001 From: Wang Xinyu Date: Mon, 17 Feb 2020 19:05:51 +0800 Subject: [PATCH 1/2] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 08052df..d628736 100644 --- a/README.md +++ b/README.md @@ -32,7 +32,7 @@ Following models are implemented, each one also has a readme inside. |[inception](./inceptionv3)| Inception v3 | |[mnasnet](./mnasnet)| MNASNet with depth multiplier of 0.5 from the paper | |[mobilenet](./mobilenetv2)| MobileNet V2 | -|[resnet](./resnet)| resnet-18 and resnet-50 are implemented | +|[resnet](./resnet)| resnet-18, resnet-50 and resnext50-32x4d are implemented | |[shufflenet](./shufflenetv2)| ShuffleNetV2 with 0.5x output channels | |[squeezenet](./squeezenet)| SqueezeNet 1.1 model | |[vgg](./vgg)| VGG 11-layer model | From 1968ed992f05c2813bd3e010c858ae5da34d5239 Mon Sep 17 00:00:00 2001 From: Wang Xinyu Date: Mon, 17 Feb 2020 19:15:37 +0800 Subject: [PATCH 2/2] Update README.md --- README.md | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index d628736..63a5655 100644 --- a/README.md +++ b/README.md @@ -8,17 +8,12 @@ TensorRTx has a brother project [Pytorchx](https://github.com/wang-xinyu/pytorch ## Test Environment -Jetson TX1 +1. Jetson TX1 / Ubuntu16.04 / cuda9.0 / cudnn7.1.5 / tensorrt4.0.2 / nvinfer4.1.3 -Ubuntu16.04 +2. GTX1080 / Ubuntu16.04 / cuda10.0 / cudnn7.6.5 / tensorrt7.0.0 / nvinfer7.0.0 -cuda9.0 +Currently, TX1 ans x86 GTX1080 were tested. trt4 api were using, some api are deprecated in trt7, but still can compile successfully. -cudnn7.1.5 - -tensorrt4.0.2/nvinfer4.1.3 - -Currently, I only test it on TX1, But I think it will be totally OK on TX2 with same software version. And also it will be easy be ported to x86. ## Models