Ubuntu 配置 NVIDIA GPU环境开展深度神经网络研究

在 Ubuntu 虚拟机 (VM) 上安装和配置 NVIDIA 图形处理单元 (GPU) 可能有点挑战，尤其是对于那些不熟悉此过程的人。但是，只要有一点耐心并采取正确的步骤，您就可以立即启动并运行 NVIDIA GPU。在本文中，我们将介绍在 Ubuntu 机器上为深度学习模型安装和配置 NVIDIA GPU 的过程，涵盖从下载必要的驱动程序到设置 GPU 以供使用的所有内容。

为了安装和配置VM环境以检测您的GPU并使GPU与CUDA工具包一起工作，可以按照以下步骤操作：

第 1 步：检查并卸载现有驱动程序

在为您的 NVIDIA GPU 安装新驱动程序之前，请务必检查您的 Ubuntu 计算机上是否已有现有驱动程序。打开终端窗口并键入以下命令：lspci | grep -i nvidia

此命令将查明 VM 上是否存在任何 NVIDIA 图形设备。如果有任何列出的，请使用以下命令卸载驱动程序和 CUDA：

sudo apt-get --purge remove "*cublas*" "cuda*" "nsight*"
sudo apt-get --purge remove "*nvidia*"

如果您通过源文件安装了 CUDA，请查找它的安装位置（例如 /usr/local）并使用以下命令将其删除：

sudo rm -rf /usr/local/cuda*

然后转到包含对 NVIDIA 存储库的引用的行

sudo vim /etc/apt/sources.list

并将其注释掉

# deb http://developer.download