Yolov5 Windows10环境配置

(1)NVIDIA显卡驱动程序下载

NVIDIA - 驱动下载

(2)CUDA下载

1.CUDA == 10.2 

CUDA Toolkit 10.2 Download | NVIDIA Developer

2.验证CUDA安装是否成功

        首先重启电脑

        然后在命令行输入`nvcc -V` ,显示如下为成功

Yolov5 Windows10环境配置

(3)CuDNN下载

1. 首先要在官网注册一个账号,如果出现“验证程序加载失败”问题,解决方法如下:

英伟达账户一直登陆不进去解决办法_du2005023029的博客-CSDN博客_nvidia验证程序加载失败请检查浏览器

2. CuDNN for CUDN10.2

https://developer.nvidia.com/rdp/cudnn-download

3. 配置路径
我的电脑——>属性——>高级系统设置——>环境变量
在系统变量的Path中,双击打开添加两个路径:

一是cudnn的bin路径`C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\cudnn\bin`

二是`C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v10.0\extras\CUPTI\libx64`

(4)Pytorch下载

Previous PyTorch Versions | PyTorch

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

(5)验证

import torch 
print(torch.cuda.is_available())

返回True代表配置环境成功

注:本人安装完numpy版本有问题,重新安装一下numpy就好了

上一篇:windows10 cmd窗口输出卡住(看这篇就够了)


下一篇:Windows10搭建opengrok服务器