TX2 运行时报错，锁业内存：terminate called after throwing an instance of 'thrust::system::system_error' what(): radix_sort: failed on 2nd step: cudaErrorHostMemoryNotRegistered: pointer does not correspond to a registered memory region Aborted (core dumped) about tensorrt-alpha HOT 3 CLOSED

@lyb36524
If the code is running in the TX2 device, please modify the code(

#if 1 // for Nvidia TX2
cv::Mat img_fp32 = cv::Mat::zeros(imgsBatch[0].size(), CV_32FC3); // todo
float* pi = m_input_src_device;
for (size_t i = 0; i < imgsBatch.size(); i++)
{
std::vector img_vec = std::vector(imgsBatch[i].reshape(1, 1));
imgsBatch[i].convertTo(img_fp32, CV_32FC3);
checkRuntime(cudaMemcpy(pi, img_fp32.data, sizeof(float) * 3 * m_param.src_h * m_param.src_w, cudaMemcpyHostToDevice));
pi += 3 * m_param.src_h * m_param.src_w;
}
#endif
}

from tensorrt-alpha.

from tensorrt-alpha.

This part of the code has been modified to be compatible with TX2.

from tensorrt-alpha.