Centos 7.6 安装GPU(RTX 3080)

在本文中，我们将详细探讨如何在CentOS 7.6环境下安装NVIDIA RTX 3080 GPU的驱动程序，CUDA工具包以及cuDNN库。这些组件对于深度学习和高性能计算至关重要，因为它们能充分利用GPU的计算能力。 ### 1. 安装基础环境 #### 1.1 安装GCC 确保系统中已经安装了GCC编译器，因为CUDA安装过程中需要它。可以使用以下命令检查GCC是否已安装： ```bash gcc --version ``` 如果没有安装，可以通过执行以下命令来安装： ```bash sudo yum install -y gcc-c++ ``` #### 1.2 安装kernel开发包 确保系统已安装最新的内核开发包，以便支持NVIDIA驱动。运行以下命令进行安装： ```bash sudo yum install -y kernel-devel ``` ### 2. 查看显卡信息 在安装驱动之前，我们需要确认系统识别到了RTX 3080显卡。通过以下命令查看系统上的所有设备： ```bash lspci | grep -i nvidia ``` 如果成功识别到显卡，你应该能看到与NVIDIA相关的输出。 ### 3. 安装显卡驱动 #### 3.1 下载显卡驱动 访问NVIDIA官方网站，找到适用于RTX 3080的最新Linux驱动程序，并下载到本地。记住保存路径。 #### 3.2 安装驱动 在非图形化界面下，切换到root用户： ```bash sudo su - ``` 然后，禁用 nouveau 驱动（NVIDIA开源驱动）： ```bash echo "blacklist nouveau" | tee /etc/modprobe.d/nouveau.conf echo "options nouveau modeset=0" | tee -a /etc/modprobe.d/nouveau.conf dracut --force ``` 重启系统后，加载新配置： ```bash reboot ``` 现在，进入图形化界面，安装下载的NVIDIA驱动。例如，如果驱动文件名为`NVIDIA-Linux-x86_64-460.32.03.run`，则运行： ```bash ./NVIDIA-Linux-x86_64-460.32.03.run -no-opengl-files ``` 按照提示进行安装。 #### 3.3 查看安装的版本 安装完成后，验证驱动版本： ```bash nvidia-smi ``` 这将显示NVIDIA驱动的版本和其他相关信息。 ### 4. 安装CUDA #### 4.1 下载CUDA 访问NVIDIA官方网站，选择适用于CentOS 7的CUDA版本，下载并解压到合适的位置。 #### 4.2 安装CUDA 运行以下命令接受许可协议： ```bash sudo bash cuda_11.2.2_460.32.03_linux.run --accept-license ``` 接着，选择自定义安装并取消选择NVIDIA驱动（因为我们已经在上一步单独安装了）。然后，按照提示完成安装。 安装完成后，添加CUDA到系统路径： ```bash echo 'export PATH=/usr/local/cuda/bin${PATH:+:$PATH}' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}' >> ~/.bashrc source ~/.bashrc ``` 验证CUDA安装： ```bash nvidia-smi cuda-gdb --version ``` ### 5. 安装cuDNN #### 5.1 下载cuDNN 在NVIDIA官网上下载cuDNN对应CUDA版本的压缩包，将其解压到一个临时目录。 #### 5.2 安装cuDNN 将cuDNN的库文件复制到CUDA安装目录： ```bash sudo cp -P include/cudnn.h /usr/local/cuda/include/ sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* ``` 更新库链接： ```bash sudo ldconfig ``` 验证cuDNN安装： ```bash cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 ``` 以上步骤将引导你完成在CentOS 7.6上为NVIDIA RTX 3080安装驱动、CUDA和cuDNN的全过程。完成后，你的系统将准备就绪，可以用于深度学习和高性能计算任务。记得在实际操作中，根据实际情况调整命令和版本号。