安装 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 安装,它会自动完成所有必要的配置。
Linux 系统安装 CUDA Toolkit
- 前提条件:
- 确保你的计算机上已经安装了 NVIDIA 显卡,并且它支持 CUDA。
- 确保你安装了适合你的显卡的 NVIDIA 驱动程序。
- 下载 CUDA Toolkit:
- 访问 NVIDIA CUDA Toolkit 下载页面。
- 选择你操作系统的版本(Linux)、版本号(比如 11.x)等,点击 Download。
- 安装 CUDA Toolkit:
- 解压下载的安装包到指定目录(例如
/opt/cuda
)。 - 按照安装脚本中的指示进行配置和安装。通常需要运行以下命令:
sudo ./cuda_xxx.run --silent --toolkit
- 安装完成后,将 CUDA 库路径添加到系统的环境变量中,编辑
~/.bashrc
或 ~/.zshrc
文件,添加以下行:
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc
通过以上步骤,你应该能够成功安装和配置 CUDA Toolkit。如果在安装过程中遇到问题,可以参考 NVIDIA 官方文档或相关社区论坛寻求帮助。