CUDA Toolkit 安装

CUDA 是一个并行计算平台和编程模型，可以利用 NVIDIA GPU 的强大计算能力来加速科学和工程计算。CUDA Toolkit 是用于开发 CUDA 应用程序的一套工具和库。

安装步骤

1. 查看系统支持的 NVIDIA GPU

   在 CUDA 官网查看自己的 GPU 是否支持 CUDA。

2. 下载安装包

   在 CUDA 官网下载适合自己 GPU 的安装包。选择 Linux 或 Windows 系统下适合的版本。

3. 安装 CUDA Toolkit

   对于 Linux 系统，可以通过命令行进行安装：

   sudo sh cuda_<version>.run
   

   对于 Windows 系统，双击安装包进行安装。

4. 配置环境变量

   添加 CUDA 和 cuDNN 的路径到环境变量。对于 Linux 系统，需要在 ~/.bashrc 或 ~/.zshrc 文件中添加下面的几行指令：

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

   对于 Windows 系统，需要设置环境变量 PATH 和 CUDA_PATH。

验证安装

1. 执行 nvcc -V 命令

   如果返回类似于下面的信息，则表明 CUDA Toolkit 已经安装成功：

   nvcc: NVIDIA (R) Cuda compiler driver
   ...
   

2. 执行 nvidia-smi 命令

   如果返回类似于下面的信息，则表明 GPU 驱动已经安装成功：

   +-----------------------------------------------------------------------------+
   | NVIDIA-SMI ...
   

总结

通过上述步骤，我们可以成功安装 CUDA Toolkit 并验证安装结果。安装成功后，我们就可以开始使用 CUDA 来加速我们的科学和工程计算任务。