广州学校论坛网站建设天津网站建设公司

ubuntu20.04，安装tensorRT, 执行example里面的./sample_char_rnn程序，测试时候报了如标题的一个错误，居然如下两行代码这样解决了，这两行命令好像是重新加载nvidia内核模块，有点玄学：

sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm

错误详情：

TensorRT-8.6.1.6/bin$ ./sample_char_rnn
&&&& RUNNING TensorRT.sample_char_rnn [TensorRT v8601] # ./sample_char_rnn
[11/26/2023-22:48:00] [I] Building and running a GPU inference engine for Char RNN model...
[11/26/2023-22:48:00] [I] Done reading weights from file...
[11/26/2023-22:48:00] [W] [TRT] Unable to determine GPU memory usage
[11/26/2023-22:48:00] [W] [TRT] Unable to determine GPU memory usage
[11/26/2023-22:48:00] [I] [TRT] [MemUsageChange] Init CUDA: CPU +0, GPU +0, now: CPU 21, GPU 0 (MiB)
[11/26/2023-22:48:00] [W] [TRT] CUDA initialization failure with error: 999. Please check your CUDA installation:  http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
&&&& FAILED TensorRT.sample_char_rnn [TensorRT v8601] # ./sample_char_rnn