ubuntu虚拟机如何使用GPU加速

在Ubuntu虚拟机中使用GPU加速，通常需要安装NVIDIA显卡驱动、CUDA Toolkit以及相应的深度学习库如cuDNN。以下是详细的步骤：

安装NVIDIA显卡驱动

1. 打开终端，输入以下命令检查NVIDIA显卡型号和推荐的驱动程序：

ubuntu-drivers devices

2. 根据终端输出，安装合适的显卡驱动。例如，安装NVIDIA 525版本的驱动程序：

sudo apt install nvidia-driver-525

3. 安装完成后重新启动电脑。

安装CUDA Toolkit

1. 运行以下命令查看支持的CUDA版本：

nvidia-smi

2. 根据推荐的CUDA版本进行下载。可以在NVIDIA CUDA Toolkit Archive找到适合的CUDA版本。

3. 下载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

4. 安装完成后，运行以下命令确认安装成功：

nvcc -V

安装cuDNN库

1. 访问NVIDIA cuDNN官方下载地址，下载对应版本的压缩包。

2. 解压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*

3. 使用以下命令查看cuDNN是否配置完成：

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

启用3D加速（适用于VirtualBox）

1. 在VirtualBox中，选择Ubuntu虚拟机，进入设置。

2. 在显示设置中，选择“VMSVGA”作为“显卡控制器”。

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

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

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

亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>