当前位置: 首页 > news >正文

Yolo环境搭建(深度学习基础环境)

需要安装的东西

  1. CUDA
  2. cuDnn
  3. 魔法

一、CUDA安装(Windows10环境)

第一:下载驱动

第二:查看显卡支持的最高CUDA的版本,以便下载对应的CUDA安装包

第三:确定CUDA版本对应的cuDNN版本,这个其实不用太关注,因为在cudnn的下载页面会列出

1、下载驱动

下载连接:驱动下载

手动驱动搜索图
手动驱动搜索图

第一行:选择游戏显卡

第二行:显卡系别(Notebooks为笔记本)

第三行:详细显卡信号

第四行:系统环境

2、查看支持的cuda版本

 快捷键:win+r 打开命令行

输入以下命令:

nvidia-smi
运行结果图

主要关注如上三部分:显卡支持的最高版本(可以比这个低)

3、下载CUDA

如下我将演示11.6版本的下载安装 

下载链接: 11.6

11.6选项图

第一行:选择操作系统

第二行:选择位数

第三行:选择操作系统版本

第四行:选择本地下载 

4、CUDA安装

需要记住安装路径,然后 无脑安装到这一步

把CUDA选一下,还需要注意一点,如果 你前面没有安装vs,直接安装的这个,需要把CUDA里面的Visual Studio Integration取消勾选,否则会安装不成功 

然后继续无脑安装

5、检查是否安装成功

快捷键:win+r 打开命令行

nvcc -V

 出现这个,说明安装成功了

二、下载cuDNN

下载链接:cudnn下载

下载需要注册账户,你们自行注册登录,成功后可以直接下载了,这里不过多阐述

下载得到:cudnn-windows-x86_64-8.4.0.27_cuda11.6-archive.zip

解压

把这三个文件夹的文件分别拷贝到CUDA安装目录对应的(bin、include、lib)文件夹中即可。

三、下载安装魔法

下载路径:一元机场icon-default.png?t=N7T8https://xingjiabijichang.com/#/knowledge

 因为文件不是很大,选择一元是最划算的

 网站中有说明文档,自行观看学习就行了,不过多阐述了

四、torch安装

 1、创建新环境

在YOLO源码文件夹中打开命令行

 输入以下命令

conda create -n YOLO python=3.9

解释:conda 创建 -n 自定义的环境名称(我这里用的是YOLO命名) 选择python版本

途中需要输入Y

创建完成图片

2、激活新环境

activate YOLO

可以看到括号中的YOLO,说明已经进入自定义的YOLO环境当中

3、新环境中安装必要的库 

首先下载YOLO所需要的库文件

pip install -r .\requirements.txt
下载视图

 等待安装完成之后,输入以下指令查看安装的库

pip list

 

发现有torch,但不是GPU版本的,使用需要删除他

pip uninstall torch torchvision torchaudio

过程中输入Y即可

4、下载torch (先必须魔法)

输入指令:

pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 -f https://download.pytorch.org/whl/torch_stable.html  -i https://pypi.tuna.tsinghua.edu.cn/simple

 安装完成之后,输入:

python

然后继续输入以下命令

import torch
from torch.backends import cudnnx = torch.Tensor([1.0])
xx = x.cuda()
print("torch版本:", torch.__version__)
print("torch_cudatoolkit版本:", torch.version.cuda)
print("torch_cuda_可用:", torch.cuda.is_available())
print("torch_cuda_计算:", xx)
print("torch_cudnn_可用:", cudnn.is_acceptable(xx))


输出结果为True,即为安装成功!

 >>> print("torch_cudnn_可用:", cudnn.is_acceptable(xx))
torch_cudnn_可用: True

 五、运行YOLO代码查看结果

1、选择刚刚创建的环境(YOLO)

2、打开detect.py文件,点击运行

3、运行完毕,结果保存在了exp4中 

六、常见问题

当安装完成好torch后,引用时,发现还是无法找到模块,并提醒dll文件有问题

解决方法:

1、下载dll文件

下载链接:文件下载 

得到如上文件,将这个文件导入自定义的YOLO环境路径中\Lib\site-pages\torch\lib中 

2、将dll文件移入环境中

在命令行中输入

conda info -e

可以看到你创建的环境以及路径

 将文件放入这个目录中

 然后就成功了!

相关文章:

  • 北京网站建设多少钱?
  • 辽宁网页制作哪家好_网站建设
  • 高端品牌网站建设_汉中网站制作
  • 利用优先级队列的堆排序练习
  • visual studio 2005 ( vs2005 , vc2005 ) 编译的应用程序无法运行的解决方案
  • 与PC1显著相关的基因 | p值计算
  • 个人旅游网(1)——数据库表详解
  • JVM1-初识JVM
  • 【cocos creator】养成游戏简易事件系统,每日随机事件,每日行动点重置,根据数据检测多结局
  • 【Unity输入】Input Manager 和 Input System对比
  • 实训第三十二天(学习playbook-roles,脚本创建数据库和表,mycat读写分离)
  • 2024年程序员金九银十面试宝典持续更新中.....
  • 【Spring Boot 3】【Web】同时启用 HTTP 和 HTTPS
  • 命令模式与宏命令:批量操作的高效实现
  • 探索Edge-TTS与WebSocket集成:打造实时语音交互系统
  • 【网络编程通关之路】 Tcp 基础回显服务器(Java实现)及保姆式知识原理详解 ! ! !
  • addroutes和next()导致的页面无法跳转问题,如登录之后无法跳转到首页,无法重定向,使用next(to)
  • 树、二叉树
  • [分享]iOS开发-关于在xcode中引用文件夹右边出现问号的解决办法
  • 【跃迁之路】【733天】程序员高效学习方法论探索系列(实验阶段490-2019.2.23)...
  • 2017届校招提前批面试回顾
  • 345-反转字符串中的元音字母
  • co模块的前端实现
  • JS基础篇--通过JS生成由字母与数字组合的随机字符串
  • Spring Boot快速入门(一):Hello Spring Boot
  • Vim Clutch | 面向脚踏板编程……
  • 近期前端发展计划
  • 跨域
  • 入手阿里云新服务器的部署NODE
  • 一天一个设计模式之JS实现——适配器模式
  • 走向全栈之MongoDB的使用
  • 做一名精致的JavaScripter 01:JavaScript简介
  • 3月27日云栖精选夜读 | 从 “城市大脑”实践,瞭望未来城市源起 ...
  • raise 与 raise ... from 的区别
  • 关于Kubernetes Dashboard漏洞CVE-2018-18264的修复公告
  • 你学不懂C语言,是因为不懂编写C程序的7个步骤 ...
  • ​渐进式Web应用PWA的未来
  • ​软考-高级-系统架构设计师教程(清华第2版)【第9章 软件可靠性基础知识(P320~344)-思维导图】​
  • # 20155222 2016-2017-2 《Java程序设计》第5周学习总结
  • # 利刃出鞘_Tomcat 核心原理解析(八)-- Tomcat 集群
  • (12)Linux 常见的三种进程状态
  • (C#)Windows Shell 外壳编程系列4 - 上下文菜单(iContextMenu)(二)嵌入菜单和执行命令...
  • (二)七种元启发算法(DBO、LO、SWO、COA、LSO、KOA、GRO)求解无人机路径规划MATLAB
  • (二)原生js案例之数码时钟计时
  • (附源码)SSM环卫人员管理平台 计算机毕设36412
  • (十八)devops持续集成开发——使用docker安装部署jenkins流水线服务
  • (四)Linux Shell编程——输入输出重定向
  • (算法)Travel Information Center
  • .[hudsonL@cock.li].mkp勒索加密数据库完美恢复---惜分飞
  • .NET CF命令行调试器MDbg入门(三) 进程控制
  • .NET Core WebAPI中使用swagger版本控制,添加注释
  • .net dataexcel winform控件 更新 日志
  • .NET 依赖注入和配置系统
  • .Net6支持的操作系统版本(.net8已来,你还在用.netframework4.5吗)
  • .NET8 动态添加定时任务(CRON Expression, Whatever)
  • .net和jar包windows服务部署
  • .NET基础篇——反射的奥妙
  • .vimrc php,修改home目录下的.vimrc文件,vim配置php高亮显示