Y9000P + ubuntu22.04深度学习环境的搭建:英伟达驱动+CUDA11.8+cuDNN8.60+ TensorRT8.5.3.1的配置过程

2023-12-20 07:58:42

一. NVIDIA驱动安装

使用命令:ubuntu-drivers devices 获取可用驱动信息

系统设置-->软件和更新-->附加驱动-->应用更改。

当然也可以采用命令安装或者到NVIDIA官网下载手动安装

驱动装好后在终端执行:nvidia-smi 如果出现如下图则安装成功

二.CUDA的安装

CUDA Toolkit Archive | NVIDIA Developer 官网下载cuda11.8,按步骤进行安装,注意在安装cuda时取消驱动安装这个选项

按照上图中选择自己对应的选项,然后执行提供的两条安装命令即可(在主目录下创建cuda文件夹,然后把下载好的安装包复制到cuda 文件夹下,然后打开终端,进入cuda文件夹,

执行

sudo sh?cuda_11.8.0_520.61.05_linux.run

)

以上命令中的版本号要根据你选择的版本修改,依次执行以上命令

如果出现以下界面,提醒你已经安装了driver的时候,直接continue。

经过一小会的卡顿后,选择accept

当选择安装内容的时候,务必把driver前面的x取消掉,按回车键取消勾选,因为我们已经安装了驱动!!!,否则会出现[ERROR]: Install of driver component failed.的情况

之后选择install便可完成cuda的安装,安装成功如下图所示

配置环境变量

修改环境变量文件

sudo gedit ~/.bashrc

将下面代码,写入文件最下方

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

然后保存退出,执行如下命令

source ~/.bashrc

此时再运行nvcc -V,即可发现cuda已安装成功:

三.cuDNN的安装

NVIDIA官网cuDNN下载页面下载cuDNN8.6.0的对应版本压缩文件夹。

在主目录创建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

到此cuDNN安装完成,验证cuDNN,查看版本

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

可以看到是8.6.0版本

四.TensorRT安装

NVIDIA官网TensorRT下载页面下载TensorRT的对应版本压缩文件夹。

Log in | NVIDIA Developer

同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

打开配置文件

sudo gedit ~/.bashrc

?把如下代码添加到配置文件最末端

export LD_LIBRARY_PATH=/data/TensorRT-8.5.3.1/lib:$LD_LIBRARY_PATH
export LIBRARY_PATH=/data/TensorRT-8.5.3.1/lib::$LIBRARY_PATH

source配置文件

source ~/.bashrc

TensoRT安装测试

TensorRT解压文件夹中找到sample目录下的sampleOnnxMNIST,在该目录中make一下,并在生成bin目录下执行:./sample_onnx_mnist。(注意图片中的文件路径)

至此上述环境配置完成,后面在装pytorch深度学习框架

文章来源:https://blog.csdn.net/LSG_Down/article/details/135077947
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。