Windows10使用cuda加速YOLOv5训练

查看自己主机的显卡信息,比如我的

显卡是:NVIDIA GeForce GTX 1060 6GB

需要安装的工具列表:

• 显卡驱动 (v465.89):
• cuda v11.3.1
• cuDNN v8.2.1
• pytorch v1.10.1

步骤一:安装显卡驱动

使用下面的链接查找cuda版本对应的显卡驱动(此步骤千万不要弄错了)

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/

NVIDIA 驱动程序下载 - 高级搜索

 

查看驱动版本相关信息 nvidia-smi

$nvidia-smi

 步骤二:安装cuda toolkit 11.3.1版本

官网:  

https://developer.nvidia.cn/zh-cn/cuda-toolkit

 

• 设置环境变量NUMBA_ENABLE_CUDASIM

 步骤三:conda安装cudatoolkit(11.3.1)

$conda install -c cudatoolkit=11.3

步骤四:安装cuDNN

官网:https://developer.nvidia.cn/rdp/cudnn-archive

 

 点击下载，然后会得到一个压缩包，将压缩包解压出来的文件全部复制到之前装的CUDA路径下的里面就可。

 

步骤五:安装对应的Pytorch 

官网:

https://pytorch.org/get-started/locally/

$conda install pytorch torchvision cudatoolkit=11.3.1 -c pytorch

 成功时显示:

 查看pytorch版本

$python
>>> import torch
>>> print(torch.__version__)
 1.10.1

判断是否能使用cuda

$python
>>> import torch
>>> torch.cuda.is_available()
True

查看cuda版本

$nvcc --version

参考:

https://www.cnblogs.com/obarong/p/14833845.html

成功时训练打印: