linux及windows安装NVIDIA CUDA并配置GPU计算环境

安装CUDA需要多个步骤，以下是详细指南：

---

1. 确认系统兼容性

• 操作系统：确保系统是CUDA支持的版本（如Ubuntu 20.04/22.04、Windows 10/11等）。
• GPU型号：确认GPU支持CUDA，可查阅NVIDIA CUDA GPUs列表。
• 驱动版本：确保安装了NVIDIA驱动，且版本与CUDA兼容。

---

2. 安装NVIDIA驱动

如果未安装NVIDIA驱动，需先安装。

Linux（Ubuntu）

1. 更新系统：

   sudo apt update
   sudo apt upgrade -y
   

2. 安装驱动：
   • 推荐使用ubuntu-drivers工具自动安装：

     sudo apt install ubuntu-drivers-common
     sudo ubuntu-drivers autoinstall
     

   • 或手动安装：

     sudo apt install nvidia-driver-<version>
     

3. 重启系统：

   sudo reboot
   

Windows

1. 访问NVIDIA驱动下载页面。
2. 选择GPU型号和操作系统，下载并安装驱动。
3. 安装完成后重启系统。

---

3. 下载CUDA Toolkit

1. 访问CUDA Toolkit下载页面。
2. 选择操作系统、架构、版本和安装方式（如Linux使用runfile或deb，Windows使用exe）。

---

4. 安装CUDA Toolkit

Linux（Ubuntu）

1. 使用deb安装：

   • 下载.deb文件后运行：

     sudo dpkg -i cuda-repo-<distro>_<version>_amd64.deb
     sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/<distro>/x86_64/7fa2af80.pub
     sudo apt update
     sudo apt install cuda
     

2. 使用runfile安装：

   • 下载.run文件后运行：

     sudo sh cuda_<version>_linux.run
     

   • 按照提示完成安装。

3. 添加环境变量：
   在~/.bashrc或~/.zshrc中添加：

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

   然后加载配置：

   source ~/.bashrc
   

Windows

1. 运行下载的.exe文件。
2. 选择安装路径，推荐默认路径。
3. 安装完成后，将CUDA的bin和libnvvp目录添加到系统环境变量：
   • 右键“此电脑” -> “属性” -> “高级系统设置” -> “环境变量”。
   • 在Path中添加：

     C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\<version>\bin
     C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\<version>\libnvvp
     

---

5. 验证CUDA安装

Linux

1. 检查CUDA版本：

   nvcc --version
   

2. 运行示例程序：

   cd /usr/local/cuda/samples/1_Utilities/deviceQuery
   sudo make
   ./deviceQuery
   

   如果显示GPU信息，说明安装成功。

Windows

1. 打开命令提示符，检查CUDA版本：

   nvcc --version
   

2. 运行示例程序：
   • 打开CUDA示例目录（默认在C:\ProgramData\NVIDIA Corporation\CUDA Samples）。
   • 编译并运行deviceQuery示例，确认GPU信息。

---

6. 安装cuDNN（可选）

如果需要深度学习支持，安装cuDNN：

1. 下载cuDNN：cuDNN下载页面。
2. 解压并复制文件到CUDA安装目录：

   tar -xzvf cudnn-<version>-linux-x64-v<version>.tgz
   sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
   sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
   

---

7. 常见问题

• 驱动冲突：确保NVIDIA驱动与CUDA版本兼容。
• 环境变量未生效：检查PATH和LD_LIBRARY_PATH是否正确设置。
• GPU未识别：运行nvidia-smi确认GPU是否被识别。