ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

Yolov5 Windows10环境配置

2021-12-09 16:03:03  阅读:288  来源: 互联网

标签:Yolov5 10.2 验证 配置 Windows10 cudnn CUDA NVIDIA 下载


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

NVIDIA - 驱动下载

(2)CUDA下载

1.CUDA == 10.2 

CUDA Toolkit 10.2 Download | NVIDIA Developer

2.验证CUDA安装是否成功

        首先重启电脑

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

(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就好了

标签:Yolov5,10.2,验证,配置,Windows10,cudnn,CUDA,NVIDIA,下载
来源: https://blog.csdn.net/qq_44144025/article/details/121819344

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有