温馨提示×

ubuntu虚拟机GPU加速支持情况

小樊
57
2025-07-17 16:12:00
栏目: 云计算

在Ubuntu虚拟机中启用GPU加速可以显著提升图形处理能力和整体性能,尤其是在运行GPU密集型应用时。以下是在Ubuntu虚拟机中启用GPU加速的详细步骤:

安装NVIDIA显卡驱动

  • 打开终端,输入以下命令检查NVIDIA显卡型号和推荐的驱动程序:

    ubuntu-drivers devices
    
  • 根据终端输出,安装合适的显卡驱动。例如,安装NVIDIA 525版本的驱动程序:

    sudo apt install nvidia-driver-525
    
  • 安装完成后重新启动电脑。

安装CUDA Toolkit

  • 运行以下命令查看支持的CUDA版本:

    nvidia-smi
    
  • 根据推荐的CUDA版本进行下载。可以在NVIDIA CUDA Toolkit Archive找到适合的CUDA版本。

  • 下载CUDA安装包并运行安装命令。例如,下载并安装CUDA 12.0:

    wget https://developer.download.nvidia.com/compute/cuda/12.0.1/local_installers/cuda_12.0.1_525.85.12_linux.run
    sudo sh cuda_12.0.1_525.85.12_linux.run
    
  • 安装完成后,运行以下命令确认安装成功:

    nvcc -V
    

安装cuDNN库

  • 访问NVIDIA cuDNN官方下载地址,下载对应版本的压缩包。

  • 解压tar包,并将文件拷贝到指定位置。例如:

    tar -xvf cudnn-linux-$arch-8.x.x.x_cudaX.Y-archive.tar.xz
    cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
    cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
    chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
    
  • 使用以下命令查看cuDNN是否配置完成:

    cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
    

启用3D加速(适用于VirtualBox)

  • 在VirtualBox中,选择Ubuntu虚拟机,进入设置。

  • 在显示设置中,选择“VMSVGA”作为“显卡控制器”。

  • 在“硬件加速”下勾选“启用3D加速”。

  • 打开Ubuntu虚拟机并安装最新的VirtualBox增强插件,安装后重新启动虚拟机。

验证GPU加速

  • 使用以下命令监控GPU的使用情况:

    nvidia-smi
    
  • 如果看到NVIDIA显卡的信息,说明驱动安装成功。

请注意,具体的步骤可能会因Ubuntu版本、虚拟机软件版本以及显卡型号的不同而有所差异。建议参考官方文档或教程以获取最准确的信息。

0