安装 CUDA Toolkit 的步骤如下,具体步骤根据你的操作系统(Windows/Linux/Mac)有所不同。以下是 Windows 和 Linux 系统的安装步骤:
Windows 系统安装 CUDA Toolkit
-
前提条件
- 确保你的计算机上已经安装了 NVIDIA 显卡,并且它支持 CUDA。
- 确保你安装了 适合你的显卡的 NVIDIA 驱动程序。
-
下载 CUDA Toolkit
- 访问 NVIDIA CUDA Toolkit 下载页面。
- 选择你操作系统的版本(Windows)、版本号(比如 11.x)等,点击 Download。
-
安装 CUDA Toolkit
- 运行下载的安装包,按照提示完成安装过程。
- 在安装过程中,你可以选择安装 CUDA Toolkit 和 NVIDIA驱动(如果你的驱动已经是最新的,可以跳过此项)。
- 推荐选择 Express 安装,它会自动为你配置好大部分环境。
- 在安装过程中,安装程序会提示你是否添加环境变量到系统中(一般默认会选中)。确保选中此选项。
- 运行下载的安装包,按照提示完成安装过程。
-
验证安装
- 安装完成后,打开命令提示符(cmd),输入以下命令:
nvcc --version
- 如果能看到 CUDA 版本信息,说明安装成功。
- 安装完成后,打开命令提示符(cmd),输入以下命令:
-
配置环境变量
如果你在安装时未自动配置环境变量,手动配置如下:- CUDA Toolkit 安装路径默认在
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXX.X
。 - 打开 环境变量设置:
- 右键点击 “此电脑” -> 选择 “属性”。
- 选择 高级系统设置 -> 点击 环境变量。
- 在 系统变量 中,添加以下两个路径:
- CUDA_PATH:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXX.X
- Path:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXX.X\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vXX.X\libnvvp
- CUDA_PATH:
- CUDA Toolkit 安装路径默认在
Linux 系统安装 CUDA Toolkit
-
前提条件
- 确保你的系统已经安装了 NVIDIA 显卡 并且支持 CUDA。
- 确保你安装了 适合你的显卡的 NVIDIA 驱动程序。
-
下载 CUDA Toolkit
- 访问 NVIDIA CUDA Toolkit 下载页面,选择 Linux 操作系统。
- 选择适合你 Linux 发行版的安装包(例如
.deb
或.rpm
格式)。
-
安装 CUDA Toolkit
-
对于
.deb
安装包: -
对于
.rpm
安装包:
-
-
设置环境变量
-
验证安装
- 打开终端,运行以下命令检查 CUDA 是否安装成功:
nvcc --version
- 你还可以使用
nvidia-smi
命令查看显卡信息:nvidia-smi
- 打开终端,运行以下命令检查 CUDA 是否安装成功:
安装后的一些常见操作
-
测试 CUDA 程序
- 在安装后,你可以运行一些官方提供的示例代码来验证安装是否正常。例如,进入 CUDA 示例代码目录,运行:
cd /usr/local/cuda/samples/1_Utilities/deviceQuery sudo make ./deviceQuery
- 如果输出显示你的显卡信息,并且没有报错,说明 CUDA 已成功安装。
- 在安装后,你可以运行一些官方提供的示例代码来验证安装是否正常。例如,进入 CUDA 示例代码目录,运行:
-
安装 cuDNN(深度学习库)
- 如果你要使用 CUDA 进行深度学习开发,建议安装 cuDNN。你可以从 NVIDIA cuDNN下载页面 下载与 CUDA 版本匹配的 cuDNN。
完成这些步骤后,你就可以开始使用 CUDA 进行开发了!如果遇到任何问题,随时可以提出。