?1、 NVIDIA驅(qū)動(dòng)安裝
系統(tǒng)設(shè)置->軟件和更新->附加驅(qū)動(dòng)->選擇NVIDIA驅(qū)動(dòng)->應(yīng)用更改。
2、CUDA的安裝
NVIDIA官網(wǎng)CUDA下載頁面下載CUDA11.8，按步驟進(jìn)行安裝，注意在安裝CUDA的時(shí)候取消驅(qū)動(dòng)安裝這個(gè)選項(xiàng)。
安裝完后配置環(huán)境變量，sudo gedit ~/.bashrc，添加：
export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
3、cudnn的安裝
NVIDIA官網(wǎng)cuDNN下載頁面下載cuDNN8.6.0的對應(yīng)版本壓縮文件夾。
?解壓后將對應(yīng)include和lib中的文件拷貝到cuda-11.8的include和lib64文件夾中：
sudo cp -r lib/* /usr/local/cuda-11.8/lib64
sudo cp -r include/* /usr/local/cuda-11.8/include
?3、TensorRT的安裝
?NVIDIA官網(wǎng)TensorRT下載頁面下載TensorRT的對應(yīng)版本壓縮文件夾。
?同cuDNN類似，解壓后將對應(yīng)include和lib中的文件拷貝到cuda-11.8的include和lib64文件夾中：
sudo cp -r lib/* /usr/local/cuda-11.8/lib64
sudo cp -r include/* /usr/local/cuda-11.8/include
4、安裝驗(yàn)證
TensorRT解壓文件夾中找到sample目錄下的sampleOnnxMNIST，在該目錄中make一下，并在生成bin目錄下執(zhí)行：./sample_onnx_mnist。
文章來源：http://www.zghlxwxcb.cn/news/detail-503514.html
Over。文章來源地址http://www.zghlxwxcb.cn/news/detail-503514.html
到了這里，關(guān)于Ubuntu20.04下CUDA11.8、cuDNN8.6.0、TensorRT8.5.3.1的配置過程的文章就介紹完了。如果您還想了解更多內(nèi)容，請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
