yum安装cuda

介绍

CUDA（Compute Unified Device Architecture）是由Nvidia公司推出的一种并行计算平台和编程模型，它可以加速应用程序并行处理的速度。本文将介绍如何使用yum命令来安装CUDA。

步骤

1. 首先需要查看系统中GPU型号，因为不同的GPU支持的CUDA版本是不同的。在终端输入以下命令：

   lspci | grep -i nvidia
   

   如果输出类似于01:00.0 VGA compatible controller: NVIDIA Corporation Device xxx (rev a1)的内容，其中xxx就是GPU型号。

2. 接下来添加官方的CUDA仓库，以Ubuntu为例，打开终端并输入以下命令：

   curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.105-1_amd64.deb
   sudo dpkg -i cuda-repo-ubuntu1804_10.1.105-1_amd64.deb
   sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
   sudo apt-get update
   

   其中cuda-repo-ubuntu1804_10.1.105-1_amd64.deb是CUDA的安装包，需要根据GPU型号选择相对应的安装包。第3行命令是添加官方的GPG密钥，以确保下载包的安全性。

3. 检查CUDA版本是否支持当前GPU型号，以Ubuntu为例，打开终端并输入以下命令：

   apt-cache policy cuda
   

   如果输出中CUDA的版本支持当前的GPU型号，则可以继续下一步。

4. 使用yum命令安装CUDA，以CentOS为例，在终端中输入以下命令：

   yum install cuda
   

   注意，此命令会自动安装当前系统所支持的最新CUDA版本。

5. 安装完成后，为了让CUDA生效，需要添加环境变量到全局的.bashrc文件中。打开.bashrc文件并添加以下内容：

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

   重新