Ubuntu 18.04(x86_64)安装配置 CUDA 11.6 + cuDNN v8.9.7 + TensorRT8.4.1.5(亲测可用)

1. 检查是否安装显卡驱动

lspci | grep -i nvidia

nvidia-smi

如果显示特定的界面,这表明显卡驱动已经被安装。若未安装驱动,则需通过手动方式进行安装。

首先,需要添加软件源:

执行以下命令以添加PPA源:

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

接下来,查询可用的驱动程序:

通过运行以下命令来查看可安装的驱动选项:

ubuntu-drivers devices

选择并安装最合适的驱动,通常推荐安装带有"recommended"标记的驱动,这个驱动版本号往往是最高的。

通常,建议直接安装带有"recommended"标记的驱动。但是 ,对于CUDA 11.6,至少需要安装版本为510的驱动程序。

sudo apt update
sudo apt install nvidia-driver-510

然后重启计算机!!!!!

2. CUDA 11.6

CUDA Toolkit Archive | NVIDIA Developer

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
sudo sh cuda_11.6.0_510.39.01_linux.run

配置环境

sudo gedit ~/.bashrc

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.6/lib64
export PATH=$PATH:/usr/local/cuda-11.6/bin
export CUDA_HOME=/usr/local/cuda-11.6

请特别留意文件夹路径的正确性。有时路径可能直接是 `/usr/local/cuda`,而其他时候可能包含具体的版本号。务必确认所引用的文件夹确实存在。

source ~/.bashrc

为确保CUDA正确安装,需要更新环境变量后进行验证。请确保检查的CUDA文件夹路径与之前设置的环境变量路径相匹配。

cat /usr/local/cuda/version.json

尽管通过使用 `nvidia-smi` 或 `nvcc --version` 命令可以查看CUDA的版本号,但这并不足以准确判断CUDA环境是否配置正确,因为即使CUDA环境未正确配置,这两个命令也能显示版本号。因此,依照前述方法检查CUDA版本和环境配置的正确性更为可靠。

2. cuDNN v8.9.7

安装cuDNN的步骤相对于CUDA来说更为简单,主要涉及下载相应版本的压缩包,然后将文件复制到特定目录并设置适当的权限。

访问NVIDIA官方网站下载cuDNN:

需要将这个文件(以及可能的其他相关文件)复制到CUDA的安装目录中。基于CUDA安装路径(/usr/local/cuda-11.6),以下是复制这些文件的命令:

sudo cp /home/fairlee/Downloads/cudnn-linux-x86_64-8.9.7.29_cuda11-archive/include/cudnn_version.h /usr/local/cuda-11.6/include/
sudo cp /home/fairlee/Downloads/cudnn-linux-x86_64-8.9.7.29_cuda11-archive/include/cudnn*.h /usr/local/cuda-11.6/include/
sudo cp /home/fairlee/Downloads/cudnn-linux-x86_64-8.9.7.29_cuda11-archive/lib64/libcudnn* /usr/local/cuda-11.6/lib64/

查看是否正确安装

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

3. TensorRT8.4.1.5

https://developer.nvidia.com/nvidia-tensorrt-8x-download

sudo gedit ~/.bashrc

注意放的目录

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/fairlee/Programs/TensorRT-8.4.1.5/lib
export LIBRARY_PATH=$LIBRARY_PATH:/home/fairlee/Programs/TensorRT-8.4.1.5/lib

sudo cp -rP ~/Programs/TensorRT-8.4.1.5/lib/* /usr/lib/
sudo cp -rP ~/Programs/TensorRT-8.4.1.5/include/* /usr/include/

测试 #在TensorRt目录文件夹下,cd到sample文件夹下

#在TensorRt目录文件夹下,cd到sample文件夹下
sudo make
cd ../bin
./sample_mnist

有两个Python版本安装在系统中:Python 2.7.17 和 Python 3.6.9。TensorRT的.whl文件是针对特定版本的Python编译的,所以需要选择与打算使用的Python版本对应的TensorRT wheel文件。

由于Python 2已经过时且不再推荐使用,应该使用Python 3的版本。系统中安装的是Python 3.6.9,因此应该安装与之兼容的TensorRT wheel文件:

tensorrt-8.4.1.5-cp36-none-linux_x86_64.whl
相关推荐
007php0074 小时前
linux服务器上CentOS的yum和Ubuntu包管理工具apt区别与使用实战
linux·运维·服务器·ubuntu·centos·php·ai编程
Jzin4 小时前
Ubuntu存储硬盘扩容-无脑ChatGPT方法
ubuntu·chatgpt
djykkkkkk4 小时前
ubuntu编译遇到的问题
linux·运维·ubuntu
qq_429856575 小时前
linux 查看服务是否开机自启动
linux·运维·服务器
7yewh6 小时前
Linux驱动开发 IIC I2C驱动 编写APP访问EEPROM AT24C02
linux·arm开发·驱动开发·嵌入式硬件·嵌入式
dessler6 小时前
Docker-Dockerfile讲解(三)
linux·运维·docker
KevinRay_7 小时前
命令行之巅:Linux Shell编程的至高艺术(中)
linux·运维·服务器·重定向·shell编程
程序员JerrySUN7 小时前
Yocto 项目 - 共享状态缓存 (Shared State Cache) 机制
linux·嵌入式硬件·物联网·缓存·系统架构
林农8 小时前
C05S16-MySQL高可用
linux·mysql·云计算
码中小白鼠8 小时前
Ubuntu系统部署Mysql8.0后设置不区分大小写
linux·mysql·ubuntu·adb