CUDA 是 NVIDIA 提供的一个并行计算平台和编程模型,它允许开发者利用 NVIDIA 的 GPU 进行通用计算。以下是在 AI 工具开源教程中关于 CUDA 安装的详细步骤。

系统要求

  • 操作系统:Windows、Linux 或 macOS
  • NVIDIA GPU:至少支持 CUDA 的 GPU
  • CUDA Toolkit:适用于你操作系统和 GPU 的最新版本

安装步骤

  1. 下载 CUDA Toolkit:访问 NVIDIA 官方网站 下载适用于你的操作系统和 GPU 的 CUDA Toolkit。

  2. 安装 CUDA Toolkit

    • 双击下载的安装包,按照提示进行安装。
    • 安装过程中,确保选择合适的安装路径和配置选项。
  3. 设置环境变量

    • 在 Windows 中,右键点击“此电脑”>“属性”>“高级系统设置”>“环境变量”。
    • 在系统变量中找到“Path”,点击“编辑”。
    • 在变量值末尾添加 ;<CUDA Toolkit 安装路径>\bin
    • 在 Linux 或 macOS 中,打开终端,编辑 .bashrc.zshrc 文件,添加以下行:
      export PATH=$PATH:<CUDA Toolkit 安装路径>/bin
      
    • 保存文件并重新加载配置。
  4. 验证安装

    • 打开命令行或终端,输入 nvcc --version,如果正确显示 CUDA Toolkit 版本信息,则表示安装成功。

注意事项

扩展阅读

想要了解更多关于 CUDA 的知识,可以阅读以下教程:

CUDA GPU