在 Windows 上安装和配置 NVIDIA 驱动程序、CUDA、cuDNN 和 TensorRT
在 Windows 上安装和配置 NVIDIA 驱动程序、CUDA、cuDNN 和 TensorRT
- 1. 安装 NVIDIA 图形驱动程序
- 2. 安装 CUDA Toolkit
- 3. 安装 cuDNN
- 4.安装 TensorRT
- 5. 常见问题
1. 安装 NVIDIA 图形驱动程序
首先需要安装兼容 CUDA 的 NVIDIA 驱动程序。
-
下载最新驱动:
- 访问 NVIDIA 官网,选择你的显卡型号(例如 RTX 3060),然后下载与其对应的驱动程序。
-
安装驱动程序:
- 运行下载的驱动安装包,选择自定义安装并勾选“干净安装”,确保旧的驱动程序不会造成冲突。
- 安装完成后,建议重启电脑。
2. 安装 CUDA Toolkit
CUDA Toolkit 是开发基于 NVIDIA GPU 的应用程序的核心工具,包含编译器、库和驱动接口。
-
下载 CUDA Toolkit:
- 访问 CUDA Toolkit 下载页面,选择与你的 GPU 兼容的 CUDA 版本(推荐最新的稳定版本,如 CUDA 11.x)。
-
安装 CUDA Toolkit:
- 下载后运行安装程序,选择 Express 或 Custom 安装(自定义安装允许你