centos虚拟机GPU加速 - 问答

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

一、硬件与宿主机准备

确认宿主机支持GPU直通
- 需主板支持IOMMU（Intel VT-d/AMD-Vi），并在BIOS中启用。
- 确保宿主机安装最新驱动，且虚拟机操作系统为CentOS 7/8及以上版本。
配置虚拟机硬件
- 添加GPU设备：在虚拟机设置中选择“添加PCI设备”，勾选NVIDIA/AMD显卡（需宿主机物理GPU支持）。
- 修改虚拟机配置文件（以VMware为例）：
  在.vmx文件中添加参数：
```
pciPassthru0.present = "TRUE"  
pciPassthru0.id = "<GPU_PCI_ID>"  
hypervisor.cpuid.v0 = "FALSE"  
```
  （<GPU_PCI_ID>可通过lspci | grep NVIDIA获取）。

二、虚拟机内安装GPU驱动

禁用开源驱动（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)

安装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  
```
验证驱动安装
```
nvidia-smi  # 显示GPU状态、驱动版本及CUDA信息  
```
若输出GPU型号、驱动版本及显存占用等信息，则配置成功。

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

安装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

验证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"参数绕过检测。

参考来源：

0 赞

0 踩