本文并未介绍显卡驱动和CUDA安装过程。网上有很多教程大家可以自行搜索。
1. 模型下载
模型训练框架地址:https://github.com/pjreddie/darknet 或:
git clone https://github.com/pjreddie/darknet
2. 下载显卡驱动 (不使用独立显卡可跳过2. 3. )
NVIDIA官方下载地址:官方驱动 | NVIDIA 或:
在—软件和更新—找到系统推荐的NVIDIA显卡驱动下载
3. 下载对应版本的CUDA
通过下表找到自己对应的CUDA版本 或:
官方文档查询:Release Notes :: CUDA Toolkit Documentation
CUDA Toolkit | Toolkit Driver Version | |
---|---|---|
Linux x86_64 Driver Version | Windows x86_64 Driver Version | |
CUDA 11.6 Update 2 | >=510.47.03 | >=511.65 |
CUDA 11.6 Update 1 | >=510.47.03 | >=511.65 |
CUDA 11.6 GA | >=510.39.01 | >=511.23 |
CUDA 11.5 Update 2 | >=495.29.05 | >=496.13 |
CUDA 11.5 Update 1 | >=495.29.05 | >=496.13 |
CUDA 11.5 GA | >=495.29.05 | >=496.04 |
CUDA 11.4 Update 4 | >=470.82.01 | >=472.50 |
CUDA 11.4 Update 3 | >=470.82.01 | >=472.50 |