Open Ghustwb opened 5 years ago
Asserion ‘engine’failed ICudaEngine engine = builder->buildCudaEngine( network ); assert(engine);
void TensorNet::caffeToTRTModel(const std::string& deployFile, const std::string& modelFile, const std::vector<std::string>& outputs, unsigned int maxBatchSize) { IBuilder* builder = createInferBuilder(gLogger); INetworkDefinition* network = builder->createNetwork(); ICaffeParser* parser = createCaffeParser(); parser->setPluginFactory(&pluginFactory); bool useFp16 = builder->platformHasFastFp16(); useFp16 = false; DataType modelDataType = useFp16 ? DataType::kHALF : DataType::kFLOAT; std::cout << deployFile.c_str() <<std::endl; std::cout << modelFile.c_str() <<std::endl; std::cout << "Start Parse Network"<<std::endl; const IBlobNameToTensor* blobNameToTensor = parser->parse(deployFile.c_str(), modelFile.c_str(), *network, // DataType::kHALF); DataType::kFLOAT); assert(blobNameToTensor != nullptr); std::cout << "Here : 3" <<std::endl; for (auto& s : outputs) network->markOutput(*blobNameToTensor->find(s.c_str())); builder->setMaxBatchSize(maxBatchSize); builder->setMaxWorkspaceSize(10 << 20); std::cout << "Here : 4"<< std::endl; if(useFp16) { builder->setHalf2Mode(true); } ICudaEngine* engine = builder->buildCudaEngine( *network ); std::cout << "Here : 5"<<std::endl; assert(engine); network->destroy(); parser->destroy(); gieModelStream = engine->serialize(); engine->destroy(); builder->destroy(); pluginFactory.destroyPlugin(); shutdownProtobufLibrary(); std::cout << "Parsed Finished" <<std::endl; }
build cudaEngine failed anyone help me? thanks
Asserion ‘engine’failed ICudaEngine engine = builder->buildCudaEngine( network ); assert(engine);
build cudaEngine failed anyone help me? thanks