1、 NVIDIA驱动安装
系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改。
2、CUDA的安装
NVIDIA官网CUDA下载页面下载CUDA11.8,按步骤进行安装,注意在安装CUDA的时候取消驱动安装这个选项。
安装完后配置环境变量,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官网cuDNN下载页面下载cuDNN8.6.0的对应版本压缩文件夹。
解压后将对应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官网TensorRT下载页面下载TensorRT的对应版本压缩文件夹。
同cuDNN类似,解压后将对应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、安装验证
TensorRT解压文件夹中找到sample目录下的sampleOnnxMNIST,在该目录中make一下,并在生成bin目录下执行:./sample_onnx_mnist。
文章来源:https://www.toymoban.com/news/detail-503514.html
Over。文章来源地址https://www.toymoban.com/news/detail-503514.html
到了这里,关于Ubuntu20.04下CUDA11.8、cuDNN8.6.0、TensorRT8.5.3.1的配置过程的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!