买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：苏州浪潮智能科技有限公司

摘要：本发明实施例提供了一种基于Docker原生CentOS的CUDA安装方法、装置、电子设备以及计算机可读存储介质，属于计算机领域。其中，方法包括在Linux系统下安装Docker；从Docker仓库中安装原生CentOS；在Linux系统下禁止nouveau驱动；在Linux系统下安装NVIDIA驱动；本发明实施例在Linux系统下安装CUDA驱动。实现Docker内，利用CentOS进行CUDA安装，而不必更换换物理机，不必更换操作系统，Docker内CentOS容器会继承物理机NVIDIA的驱动与CUDA程序，大大方便了CUDA开发的时间，简化了CUDA测试的流程。

主权项：1.一种基于Docker原生CentOS的CUDA安装方法，其特征是，包含步骤：在Linux系统安装Docker；从Docker仓库中安装原生CentOS；在Docker内创建CentOS容器及驱动映射；在物理机禁止nouveau驱动；在Linux系统安装NVIDIA驱动；在Linux系统安装CUDA驱动；将nvidia物理机下的驱动映射到CentOS，并将CUDA驱动及CUDAsample映射到对应目录下。

全文数据：基于docker原生centos的cuda的安装方法、装置及设备技术领域本发明涉及计算机技术领域，特别涉及基于docker原生centos的cuda的安装方法；还涉及基于docker原生centos的cuda的安装方法装置、设备以及计算机可读存储介质。背景技术Docker是一个开源的应用容器引擎，让开发者可以打包他们的应用以及依赖包到一个可移植的容器中，然后发布到任何流行的Linux机器上，也可以实现虚拟化。容器是完全使用沙箱机制，相互之间不会有任何接口。CUDAComputeUnifiedDeviceArchitecture，是显卡厂商NVIDIA推出的运算平台。CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。它包含了CUDA指令集架构ISA以及GPU内部的并行计算引擎。开发人员现在可以使用C语言来为CUDA构编写程序，C语言是应用最广泛的一种高级编程语言。所编写出的程序可以在支持CUDA处理器上以超高性能运行，CUDA主要运行在NVIDIA的显卡上在实际应用场景中，用户一般在物理机上安装了显卡驱动和CUDA，客户进行CUDA编程开发时，需要进行跨平台跨操作系统的切换测试，配置过程耗时较长，并且会占用大量资源，造成了不必要的开销和时间的浪费。CentOSCommunityEnterpriseOperatingSystem，中文意思是社区企业操作系统是Linux发行版之一，它是来自于RedHatEnterpriseLinux依照开放源代码规定释出的源代码所编译而成。由于出自同样的源代码，因此有些要求高度稳定性的服务器以CentOS替代商业版的RedHatEnterpriseLinux使用。有鉴于此，如何提供基于docker原生centos的cuda的安装与测试方案，是本领域技术人员亟待解决的技术问题。发明内容本发明的目的是提供一种基于docker原生centos下的cuda安装方法，实现docker内，利用centos进行cuda安装，而不必更换换物理机，不必更换操作系统，docker内centos容器会继承物理机NVIDIA的驱动与CUDA程序，大大方便了CUDA开发的时间，简化了CUDA测试的流程；本发明的另一目的是提供一种基于docker原生centos下的cuda安装装置、电子设备以及计算机可读存储介质，均具有上述技术效果。为解决上述技术问题，本发明提供了一种基于docker原生centos下的cuda安装方法，包括：一种基于docker原生centos下的cuda安装方法，包含步骤：在linux系统下安装docker；从docker仓库中安装原生centos；在linux系统下禁止nouveau驱动；在linux系统下安装NVIDIA驱动；在linux下系统下安装cuda驱动。可选的，cuda驱动安装完毕后，包含步骤：配置cuda环境变量；确认系统是否已正常识别NVIDIA显卡，并检测。可选的，确认系统已正常识别NVIDIA显卡，并检测后，包含步骤：在docker内创建centos容器及驱动映射；在docker容器内验证cuda程序。可选的，所述在docker容器内验证cuda程序，包含：验证cuda程序驱动是否正常；验证cuda映射是否正常；验证cuda运行是否正常。为解决上述技术问题，本发明还提供了一种基于docker原生centos下的cuda安装装置，包含：安装模块，用于在linux系统下安装docker；从docker仓库中安装原生centos；在linux系统下禁止nouveau驱动；在linux系统下安装NVIDIA驱动；在linux下系统下安装cuda驱动。配置检测模块，用于cuda驱动安装完毕后，配置cuda环境变量；确认系统已正常识别NVIDIA显卡，并检测。验证模块，用于确认系统已正常识别NVIDIA显卡，并检测后，在docker内创建centos容器及驱动映射；在docker容器内验证cuda程序。可选的，基于docker原生centos下的cuda安装装置，所述验证模块用于验证cuda程序驱动是否正常；验证cuda映射是否正常；验证cuda运行是否正常。为解决上述技术问题，本发明还提供了一种基于docker原生centos下的cuda安装设备，包括，存储器，用于存储计算机程序；处理器，用于执行所述计算机程序时实现上述的基于docker原生centos下的cuda安装方法的步骤。为解决上述技术问题，本发明还提供了一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现上述的基于docker原生centos下的cuda安装方法的步骤。本发明所提供的基于docker原生centos下的cuda安装方法，包括在linux系统下安装docker；从docker仓库中安装原生centos；在linux系统下禁止nouveau驱动；在linux系统下安装NVIDIA驱动；在linux下系统下安装cuda驱动。可见，本发明实施例所提供的基于docker原生centos下的cuda安装方法，利用docker原生的centos镜像，保证了系统的可用性和跨平台的可移植性，减少了跨平台测试需要配置物理机的操作，提高了工作效率。通过在物理机docker内安装原生centos镜像，及映射物理机CUDA到docker内镜像的方法，可以有效缩短配置时间，快速测试cuda程序，保证物理机环境的稳定性。本发明实施例所提供的基于docker原生centos下的cuda安装装置、设备以及计算机可读存储介质，具有上述技术效果。附图说明为了更清楚地说明本发明实施例中的技术方案，下面将对现有技术和实施例中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本发明的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。图1为本发明实施例所提供的基于docker原生centos下的cuda安装方法的流程示意图；图2为本发明实施例所提供的另一基于docker原生centos下的cuda安装方法的流程示意图；图3为本发明实施例所提供的另一基于docker原生centos下的cuda安装方法的流程示意图；图4为本发明实施例所提供的基于docker原生centos下的cuda安装装置的示意图；图5为本发明实施例所提供的基于docker原生centos下的cuda安装设备的示意图。具体实施方式本发明的核心是提供基于docker原生centos下的cuda安装方法，通过在物理机docker内安装原生centos镜像，及映射物理机CUDA到docker内镜像的方法，可以有效缩短配置时间，快速测试cuda程序，保证物理机环境的稳定性。本发明的另一核心是提供基于docker原生centos下的cuda安装装置、设备以及计算机可读存储介质，均具有上述技术效果。为使本发明实施例的目的、技术方案和优点更加清楚，下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。请参考图1，图1为本发明实施例所提供的基于docker原生centos下的cuda安装方法的流程示意图；参考图1，该基于docker原生centos下的cuda安装方法包括：在linux系统下安装docker；从docker仓库中安装原生centos；在linux系统下禁止nouveau驱动；在linux系统下安装NVIDIA驱动；在linux下系统下安装cuda驱动。具体的，在linux系统下安装docker，采用如下语句。[root@mu01～]#yumlocalinstall–ydocker-engine-selinux-17.05.0.ce-1.el7.centos.noarch.rpmdocker-engine-17.05.0.ce-1.el7.centos.x86_64.rpm。安装docker后，从docker仓库pull原生centos，具体采用如下语句。[root@mu01～]#dockeerpullcentos。在linux系统下禁止nouveau驱动，具体采用如下语句。[root@mu01～]#lsmod|grepnouveau。添加配置vimusrlibmodprobe.dblacklist-nouveau.conf。在文件中写入并保存如下两行：blacklistnouveauoptionsnouveaumodeset＝0。在linux系统下安装NVIDIA驱动，具体采用如下语句。shNVIDIA-Linux-x86_64-396.26.run。在linux下系统下安装cuda驱动，具体采用如下语句。.cuda_9.2.88_396.26_linux.run--silent--samples–toolkit。在上述实施例的基础上，请参考图2，图2为本发明实施例所提供的另一基于docker原生centos下的cuda安装方法的流程示意图；参考图2，该基于docker原生centos下的cuda安装方法，cuda驱动安装完毕后，包含步骤：配置cuda环境变量；确认系统已正常识别NVIDIA显卡，并检测。本实施例中，配置cuda环境变量，具体采用如下语句。exportPATH＝usrlocalcudabin${PATH:+:${PATH}}bash-c"echo'exportPATH＝usrlocalcudabin:$PATH'etcprofile"bash-c"echousrlocalcudalib64etcld.so.conf.dcuda.conf"ldconfig。本实施例中，确认系统已正常识别NVIDIA显卡，并检测，具体采用如下语句。cdusrlocalcudasamples1_UtilitiesdeviceQuerymake.deviceQuery。在上述实施例的基础上，请参考图3，图3为本发明实施例所提供的另一基于docker原生centos下的cuda安装方法的流程示意图；参考图3，该基于docker原生centos下的cuda安装方法，确认系统已正常识别NVIDIA显卡，并检测后，包含步骤：在docker内创建centos容器及驱动映射；在docker容器内验证cuda程序。本实施例中，在docker内创建centos容器及驱动映射，具体采用如下语句。dockerrun-it--devicedevnvidia0:devnvidia0–devicedevnvidia1:devnvidia1--devicedevnvidiactl:devnvidiactl–devicedevnvidia-uvm:devnvidia-uvm–vusrlocalnvidia:usrlocalnvidia-vhome:home-vusr:usr–vusrlocalcuda-9.2:usrlocalcuda-9.2–vrootNVIDIA_CUDA-9.2_Samples:rootNVIDIA_CUDA-9.2_Samplescentosbinbash。以上命令将nvidia物理机下的驱动映射到docker容器内，并将cuda驱动及cudasample映射到对应目录下。本实施例中，在docker容器内验证cuda程序，本实施例中，具体包含验证驱动是否正常；具体采用语句：nvidia-smi。验证cuda映射是否正常；具体采用语句：nvcc--version验证cuda运行是否正常；具体采用语句：cdusrlocalcudasamples1_UtilitiesdeviceQuerymake.deviceQuery。综上所述，本发明所提供的基于docker原生centos下的cuda安装方法，通过在物理机docker内安装原生centos镜像，及映射物理机CUDA到docker内镜像的方法，可以有效缩短配置时间，快速测试cuda程序，保证物理机环境的稳定性。本发明还提供了一种基于docker原生centos下的cuda安装装置，下文描述的该装置可以与上文描述的方法相互对应参照。请参考图4，图4为本发明实施例所提供一种的基于docker原生centos下的cuda安装装置的示意图；结合图4，该装置包括：安装模块，用于在linux系统下安装docker；从docker仓库中安装原生centos；在linux系统下禁止nouveau驱动；在linux系统下安装NVIDIA驱动；配置检测模块，用于cuda驱动安装完毕后，配置cuda环境变量；确认系统已正常识别NVIDIA显卡，并检测；验证模块，用于确认系统已正常识别NVIDIA显卡，并检测后，在docker内创建centos容器及驱动映射；在docker容器内验证cuda程序。在上述实施例的基础上，可选的，还包括：所述验证模块用于验证cuda程序驱动是否正常；验证cuda映射是否正常；验证cuda运行是否正常。本发明还提供了一种基于docker原生centos下的cuda安装设备，请参考图5，图5为本发明实施例所提供一种基于docker原生centos下的cuda安装的示意图；参考图5，该设备包括：存储器1，用于存储计算机程序；处理器2，用于执行所述计算机程序时实现上述基于docker原生centos下的cuda安装方法的步骤。对于本发明所提供的设备的介绍请参照上述方法的实施例，本发明在此不做赘述。本发明还提供了一种计算机可读存储介质，该计算机可读存储介质上存储有计算机程序，该计算机程序被处理器执行时实现上述基于docker原生centos下的cuda安装方法的步骤。该计算机可读存储介质可以包括：U盘、移动硬盘、只读存储器Read-OnlyMemory，ROM、随机存取存储器RandomAccessMemory，RAM、磁碟或者光盘等各种可以存储程序代码的介质。对于本发明所提供的计算机可读存储介质的介绍请参照上述方法实施例，本发明在此不做赘述。说明书中各个实施例采用递进的方式描述，每个实施例重点说明的都是与其他实施例的不同之处，各个实施例之间相同相似部分互相参见即可。对于实施例公开的装置、设备以及计算机可读存储介质而言，由于其与实施例公开的方法相对应，所以描述的比较简单，相关之处参见方法部分说明即可。专业人员还可以进一步意识到，结合本文中所公开的实施例描述的各示例的单元及算法步骤，能够以电子硬件、计算机软件或者二者的结合来实现，为了清楚地说明硬件和软件的可互换性，在上述说明中已经按照功能一般性地描述了各示例的组成及步骤。这些功能究竟以硬件还是软件方式来执行，取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能，但是这种实现不应认为超出本发明的范围。结合本文中所公开的实施例描述的方法或算法的步骤可以直接用硬件、处理器执行的软件模块，或者二者的结合来实施。软件模块可以置于随机存储器RAM、内存、只读存储器ROM、电可编程ROM、电可擦写可编程ROM、寄存器、硬盘、可移动磁盘、CD-ROM、或技术领域内所公知的任意其它形式的存储介质中。以上对本发明所提供的计算机管理方法、相关设备以及计算机可读存储介质进行了详细介绍。本文中应用了具体个例对本发明的原理及实施方式进行了阐述，以上实施例的说明只是用于帮助理解本发明的方法及其核心思想。应当指出，对于本技术领域的普通技术人员来说，在不脱离本发明原理的前提下，还可以对本发明进行若干改进和修饰，这些改进和修饰也落入本发明权利要求的保护范围。

权利要求：1.一种基于docker原生centos下的cuda安装方法，其特征是，包含步骤：在linux系统下安装docker；从docker仓库中安装原生centos；在linux系统下禁止nouveau驱动；在linux系统下安装NVIDIA驱动；在linux下系统下安装cuda驱动。2.如权利要求1所述的基于docker原生centos下的cuda安装方法，其特征是，cuda驱动安装完毕后，包含步骤：配置cuda环境变量；确认系统已正常识别NVIDIA显卡，并检测。3.如权利要求2所述的基于docker原生centos下的cuda安装方法，其特征是，确认系统已正常识别NVIDIA显卡，并检测后，包含步骤：在docker内创建centos容器及驱动映射；在docker容器内验证cuda程序。4.如权利要求3所述的基于docker原生centos下的cuda安装方法，其特征是，所述在docker容器内验证cuda程序，包含：验证cuda程序驱动是否正常；验证cuda映射是否正常；验证cuda运行是否正常。5.一种基于docker原生centos下的cuda安装装置，其特征是，包含：安装模块，用于在linux系统下安装docker；从docker仓库中安装原生centos；在linux系统下禁止nouveau驱动；在linux系统下安装NVIDIA驱动；配置检测模块，用于cuda驱动安装完毕后，配置cuda环境变量；确认系统已正常识别NVIDIA显卡，并检测；验证模块，用于确认系统已正常识别NVIDIA显卡，并检测后，在docker内创建centos容器及驱动映射；在docker容器内验证cuda程序。6.如权利要求5所述的基于docker原生centos下的cuda安装装置，其特征是，所述验证模块用于验证cuda程序驱动是否正常；验证cuda映射是否正常；验证cuda运行是否正常。7.一种基于docker原生centos下的cuda安装设备，其特征是，包括，存储器，用于存储计算机程序；处理器，用于执行所述计算机程序时实现如权利要求1至4任一项所述的基于docker原生centos下的cuda安装方法的步骤。8.一种计算机可读存储介质，其特征是，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至4任一项所述的基于docker原生centos下的cuda安装方法的步骤。

百度查询： 苏州浪潮智能科技有限公司 基于Docker原生CentOS的CUDA的安装方法、装置及设备