博客

ubuntu安装显卡驱动和cuda教程

  • Post category:Python

Ubuntu安装显卡驱动和CUDA教程

为了在Ubuntu系统上成功安装显卡驱动和CUDA，需要按照以下步骤操作：

步骤一：检查显卡型号

我们需要先检查自己的显卡型号，以确保选择正确的驱动程序版本。在终端输入以下命令：

lspci | grep -i nvidia

输出的信息中应该会包含Nvidia显卡的型号，例如Nvidia GeForce GTX 1080。

步骤二：添加Nvidia PPA

接下来，我们需要向Ubuntu的软件源列表中添加Nvidia PPA。在终端输入以下命令：

sudo add-apt-repository ppa:graphics-drivers/ppa

运行命令后，系统会询问你是否要处理添加的ppa，输入“Y”后按回车进行确认。

步骤三：安装显卡驱动

安装显卡驱动前，可以使用以下命令列出可用的显卡驱动版本：

apt search nvidia-driver

选择合适的显卡驱动版本后，使用以下命令安装：

sudo apt install nvidia-driver-<version>

这里的“”需要替换成你所选择的显卡驱动版本号。

步骤四：安装CUDA

安装完显卡驱动后，接下来我们需要安装CUDA。

首先，下载适用于你的系统版本和显卡架构的CUDA Toolkit。在终端输入以下命令下载：

wget https://developer.nvidia.com/compute/cuda/<version>/Prod/local_installers/cuda_<version>_linux.run

同样，需要将“”替换成你要下载的CUDA版本号。

接下来，运行以下命令安装CUDA Toolkit：

sudo sh cuda_<version>_linux.run

安装过程中，需要按照提示进行操作，例如同意许可协议、选择安装路径等。

步骤五：配置环境变量

安装完成后，我们需要配置环境变量，使系统能够找到CUDA Toolkit。在终端输入以下命令打开环境变量配置文件：

sudo nano /etc/profile

在文件最后添加以下内容：

export CUDA_HOME=/usr/local/cuda-<version>
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/$CUDA_HOME/lib64:$LD_LIBRARY_PATH

同样，需要将“”替换成你安装的CUDA版本号。

保存并退出文件后，使用以下命令使配置文件生效：

source /etc/profile

示例说明一：安装CUDA 10.2和Nvidia GeForce GTX 1080驱动

假设我们的显卡型号是Nvidia GeForce GTX 1080，我们想安装的CUDA版本是10.2。

第一步：

lspci | grep -i nvidia

输出：

01:00.0 VGA compatible controller: NVIDIA Corporation GP104 [GeForce GTX 1080] (rev a1)

第二步：

sudo add-apt-repository ppa:graphics-drivers/ppa

第三步：

sudo apt install nvidia-driver-440

第四步：

wget https://developer.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

第五步：

sudo nano /etc/profile

添加以下内容：

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

保存并退出文件后，使用以下命令使配置文件生效：

source /etc/profile

示例说明二：安装CUDA 11.1和Nvidia GeForce GTX 1660驱动

假设我们的显卡型号是Nvidia GeForce GTX 1660，我们想安装的CUDA版本是11.1。

第一步：

lspci | grep -i nvidia

输出：

2a:00.0 VGA compatible controller: NVIDIA Corporation TU116 [GeForce GTX 1660] (rev a1)

第二步：

sudo add-apt-repository ppa:graphics-drivers/ppa

第三步：

sudo apt install nvidia-driver-460

第四步：

wget https://developer.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run

第五步：

sudo nano /etc/profile

添加以下内容：

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

保存并退出文件后，使用以下命令使配置文件生效：

source /etc/profile

以上就是Ubuntu安装显卡驱动和CUDA的完整攻略。

Tags: