玩转AIGC：PVE环境下显卡直通配置全攻略

在人工智能生成内容(AIGC)领域，大模型的训练与推理往往需要强大的计算资源支持，尤其是图形处理能力。本文将指导您如何在Proxmox Virtual Environment (PVE)这一流行的虚拟化平台上配置显卡直通，为AI 虚拟机 提供直接的显卡访问权限，从而显著提升大模型的运行效率。

一、引言

AIGC（Artificial Intelligence Generated Content）利用人工智能技术和算法自动生成文本、图像、音频和 视频 等多种形式的内容。随着 深度学习 模型的不断完善，AIGC在各个领域的应用日益广泛。然而，大模型的训练与推理对计算资源尤其是GPU资源有着极高的要求。因此，在虚拟化环境中实现显卡直通，对于提升AIGC任务的执行效率具有重要意义。

二、环境准备

1. 硬件环境

• 服务器 ：选择支持VT-x/VT-d等虚拟化扩展的服务器，如搭载Intel E5-2683 v4 CPU的服务器。
• 内存 ：足够的内存以保证虚拟机的稳定运行，建议至少64GB DDR4内存。
• 显卡 ：选择支持直通技术的显卡，如NVIDIA P104-100等。

2. 软件环境

• 虚拟化平台 ：Proxmox Virtual Environment (PVE) 8.x 或更高版本。
• 操作系统 ：Ubuntu 22.04 LTS 或其他支持AIGC任务的操作系统。

三、配置显卡直通

1. 开启IOMMU功能

IOMMU（Input/Output Memory Management Unit）是实现显卡直通的关键。首先，需要在PVE中开启IOMMU功能。

编辑GRUB配置文件 ：打开 /etc/default/grub 文件，修改 GRUB_CMDLINE_LINUX_DEFAULT 参数，加入 intel_iommu=on iommu=pt 等选项。

更新GRUB配置 ：执行 update-grub 命令，使修改生效。

重启PVE系统 ：重启PVE以应用更改，并验证IOMMU是否成功开启。

2. 屏蔽默认显卡驱动

为防止默认显卡驱动与直通显卡冲突，需要将其加入黑名单。

编辑modprobe配置文件 ：在 /etc/modprobe.d/ 目录下创建或编辑配置文件，如 blacklist.conf ，加入 blacklist nouveau 、 blacklist nvidia 等命令，以屏蔽NVIDIA显卡的默认驱动。

3. 配置虚拟机以使用直通显卡

• 创建或编辑虚拟机 ：在PVE管理界面中创建或编辑虚拟机，将直通显卡分配给该虚拟机。
• 设置虚拟机硬件 ：确保虚拟机的CPU、内存等资源配置满足AIGC任务的需求。
• 启动虚拟机 ：启动虚拟机并验证显卡直通是否成功。

四、验证与测试

• 查看系统 日志 ：使用 dmesg 命令查看系统日志，确认IOMMU和VFIO模块已正确加载。
• 运行AIGC任务 ：在虚拟机中运行AIGC任务，如文本生成、图像渲染等，观察性能提升情况。