ubuntu虛擬機如何配置GPU加速

在Ubuntu虛擬機中配置GPU加速，通常需要以下幾個步驟：

1. 安裝NVIDIA顯卡驅動：

• 首先，確保系統已更新到最新版本：

  sudo apt update
  sudo apt upgrade
  

• 安裝NVIDIA驅動?？梢酝ㄟ^以下命令安裝推薦的驅動版本：

  sudo ubuntu-drivers autoinstall
  

• 安裝完成后，重啟系統以應用新的GPU驅動：

  sudo reboot
  

• 驗證驅動安裝是否成功，可以使用以下命令：

  nvidia-smi
  

  如果看到NVIDIA顯卡的信息，說明驅動安裝成功。

2. 安裝CUDA Toolkit：

• 訪問NVIDIA官方網站，選擇適合你的Ubuntu版本和架構的CUDA Toolkit版本進行下載。

• 例如，對于Ubuntu 20.04和CUDA 11.8，可以使用以下命令下載并安裝CUDA Toolkit：

  wget https://developer.download.nvidia.com/compute/cuda/11.8.1/local_installers/cuda_11.8.1_505.49.06_linux.run
  sudo sh cuda_11.8.1_505.49.06_linux.run --no-opengl-files --no-x-check --no-nouveau-check
  

• 安裝過程中會有幾次確認，分別確認是否安裝、是否要安裝32位庫以及是否要覆蓋xorg配置文件。選擇相應的選項進行安裝。

3. 配置環境變量：

• 編輯~/.bashrc文件，添加以下行：

  export PATH=/usr/local/cuda-11.8/bin:$PATH
  export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
  

• 保存并更新環境變量：

  source ~/.bashrc
  

4. 安裝cuDNN庫：

• 訪問NVIDIA官方網站，選擇適合你的CUDA版本的cuDNN庫進行下載。

• 例如，對于CUDA 11.8，可以使用以下命令下載并安裝cuDNN：

  wget https://developer.download.nvidia.com/compute/redist/cudnn/v8.6.5.32/cudnn-11.8-linux-x64-v8.6.5.32.tgz
  tar -xzvf cudnn-11.8-linux-x64-v8.6.5.32.tgz
  sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.8/include
  sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.8/lib64
  sudo chmod a+r /usr/local/cuda-11.8/include/cudnn*.h /usr/local/cuda-11.8/lib64/libcudnn*
  

• 更新動態鏈接庫緩存：

  sudo ldconfig
  

5. 啟用3D加速（如果使用VirtualBox）：

• 在VirtualBox的設置中，選擇“VMSVGA”作為“顯卡控制器”，并在“硬件加速”下勾選“啟用3D加速”。

完成以上步驟后，你的Ubuntu虛擬機應該已經配置好了GPU加速?？梢酝ㄟ^運行一些GPU密集型的應用程序或框架來驗證GPU加速是否生效。例如，可以使用TensorFlow或PyTorch等深度學習框架來測試GPU加速功能。