fix

src/Network.cpp

@@ -1,5 +1,4 @@
 #include <iostream>
-#include "NvInfer.h"
 
 #include "tkdnn.h"
 #include "Network.h"
@@ -12,10 +11,9 @@ Network::Network(dataDim_t input_dim) {
 
     float tk_ver = float(tkDNN::getVersion())/1000;
     float cu_ver = float(cudnnGetVersion())/1000;
-    float rt_ver = float(NV_TENSORRT_MAJOR) + float(NV_TENSORRT_MINOR)/10 + float(NV_TENSORRT_PATCH)/100;
 
     std::cout<<"New NETWORK (tkDNN v"<<tk_ver
-             <<", CUDNN v"<<cu_ver<<", TensorRT v"<<rt_ver<<")\n";
+             <<", CUDNN v"<<cu_ver<<")\n";
     dataType = CUDNN_DATA_FLOAT;
     tensorFormat = CUDNN_TENSOR_NCHW;
 
@@ -57,4 +55,4 @@ dataDim_t Network::getOutputDim() {
             return layers[num_layers-1]->output_dim;
 }
 
-}
+}