在CentOS虚拟机中实现GPU加速，需根据虚拟化平台（如VMware、ESXi等）选择合适方案，核心步骤如下：

一、硬件与宿主机准备

1. 确认宿主机支持GPU直通

   • 需主板支持IOMMU（Intel VT-d/AMD-Vi），并在BIOS中启用。
   • 确保宿主机安装最新驱动，且虚拟机操作系统为CentOS 7/8及以上版本。

2. 配置虚拟机硬件

   • 添加GPU设备：在虚拟机设置中选择“添加PCI设备”，勾选NVIDIA/AMD显卡（需宿主机物理GPU支持）。
   • 修改虚拟机配置文件（以VMware为例）：
     在.vmx文件中添加参数：

     pciPassthru0.present = "TRUE"  
     pciPassthru0.id = ""  
     hypervisor.cpuid.v0 = "FALSE"  
     

     （可通过lspci | grep NVIDIA获取）。

二、虚拟机内安装GPU驱动

1. 禁用开源驱动（Nouveau）

   # 查看是否加载Nouveau  
   lsmod | grep nouveau  
   # 禁用Nouveau（添加以下内容到/etc/modprobe.d/blacklist.conf）  
   blacklist nouveau  
   options nouveau modeset=0  
   # 重建initramfs  
   dracut -f /boot/initramfs-$(uname -r).img $(uname -r)  
   

2. 安装NVIDIA驱动与CUDA

   • 方法1：使用.run文件安装
     从NVIDIA官网下载对应CUDA版本的驱动包（如NVIDIA-Linux-x86_64-xxx.run），执行安装：

     chmod +x NVIDIA-Linux-x86_64-xxx.run  
     ./NVIDIA-Linux-x86_64-xxx.run --no-opengl-files --no-x-check  
     

     （--no-opengl-files避免与虚拟机图形界面冲突，--no-x-check跳过X服务检测）。
   • 方法2：使用YUM仓库（推荐）
     添加NVIDIA官方仓库并安装：

     yum install epel-release  
     yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo  
     yum install nvidia-driver nvidia-utils cuda  
     

3. 验证驱动安装

   nvidia-smi  # 显示GPU状态、驱动版本及CUDA信息  
   

   若输出GPU型号、驱动版本及显存占用等信息，则配置成功。

三、启用GPU加速（以AI/深度学习为例）

1. 安装CUDA Toolkit

   • 若驱动安装时未包含CUDA，需单独安装对应版本的CUDA Toolkit，并配置环境变量：

     echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc  
     echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc  
     source ~/.bashrc  
     

2. 验证GPU加速功能

   • 运行CUDA示例程序（如deviceQuery）：

     cd /usr/local/cuda/samples/1_Utilities/deviceQuery  
     make  
     ./deviceQuery  
     

     若输出“Result = PASS”且显示GPU信息，则加速功能正常。

注意事项

• 性能优化：为虚拟机分配足够CPU/内存资源，建议至少4核CPU+8GB内存。
• 驱动兼容性：确保驱动版本与CUDA、虚拟机内核版本匹配，可通过nvidia-smi查看驱动支持的CUDA版本。
• 虚拟化平台差异：
  • VMware需手动配置.vmx参数，ESXi需在硬件管理中启用直通并添加PCI设备。
  • 部分消费级显卡（如GeForce RTX系列）可能需添加pciPassthru.use64bitMMIO = "TRUE"参数绕过检测。

参考来源：