[发明专利]一种服务器CUDA环境配置方法、装置、设备及存储介质

说明书

本发明涉及一种服务器CUDA环境配置方法、装置、设备及存储介质，其特征在于，所述方法包括：利用脚本配置系统yum源；通过补充安装统一计算设备架构的依赖包配置依赖环境；下载并安装统一计算设备架构；通过脚本设置统一计算设备架构的环境变量。本发明通过利用脚本配置yum源，准备好所有CUDA所需要的软件包以及其依赖包，补充安装主要依赖包，安装好CUDA后，再利用脚本设置环境变量。整个CUDA配置过程不需要人工安装大量的软件包和依赖包，不需要人工输入大量复杂的长指令，操作简单，节省了大量时间和人力成本。

技术领域

本发明属于服务器系统开发技术领域，具体涉及一种服务器CUDA环境配置方法、装置、设备及存储介质。

背景技术

如今，AI技术和深度学习技术应用越来越广泛。数据集的不断扩大和计算模型的越来越复杂，对硬件设备提出了更严苛的需求。其中解决这个问题的就是GPU服务器。GPU服务器是基于GPU的应用于视频编解码、深度学习、科学计算等多种场景的快速、稳定、弹性的计算服务。GPU，即图形处理器(Graphics Processing Unit)，是NVIDIA厂商推出的一款硬件设备，出色的图形处理能力和高性能计算能力提供极致计算性能，有效解放计算压力，提升产品的计算处理效率与竞争力。NVIDIA厂商推出GPU通用并行计算架构，叫做CUDA(统一计算设备架构Compute Unified Device Architecture)，可以提供变成模型，使GPU加速计算，以解决更复杂的计算问题。

在服务器系统开发过程中对GPU部件测试时，会涉及到使用CUDA环境；在高性能模型测试过程中，也需要搭建CUDA环境来做应用的仿真运行。而Linux操作系统下配置CUDA环境，主要分三个步骤：1、操作系统安装必要的软件包；2、安装CUDA软件包；3、配置环境变量。这三个步骤中，所需要的软件包名称、版本以及环境变量的配置方式，相对来说都比较繁琐复杂，还需要输入大量较多的长命令。配置效率低，浪费大量的时间和人力资源。

发明内容

本发明的目的在于，针对上述现有技术存在的缺陷，提供设计一种服务器CUDA环境配置方法、装置、设备及存储介质，以解决上述技术问题。

第一方面，本申请实施例提供一种服务器CUDA环境配置方法，其特征在于，所述方法包括：

利用脚本配置系统yum源；

通过补充安装统一计算设备架构的依赖包配置依赖环境；

下载并安装统一计算设备架构；

通过脚本设置统一计算设备架构的环境变量。

结合第一方面，在第一方面的第一种实施方式中，所述利用脚本配置系统yum源包括：

根据yum名称、镜像所在位置、固定参数和rpm包所在位置编写配置脚本；

将RHEL系统的iso镜像挂载到操作系统指定文件夹下；

将所述配置脚本复制到指定文件夹下；

刷新系统yun源完成配置。

结合第一方面，在第一方面的第二种实施方式中，所述通过补充安装统一计算设备架构的依赖包配置依赖环境包括：

安装dkms-xxx.rpm；

安装libvdpau-xxx.rpm。

结合第一方面，在第一方面的第三种实施方式中，所述通过脚本设置统一计算设备架构的环境变量包括：

根据配置环境变量的指令创建环境变量脚本；

在所述环境变量脚本所在文件夹下输入执行指令使环境变量生效。

第二方面，本申请实施例提供一种服务器CUDA环境配置装置，所述装置包括：