深度学习环境搭建(pycharm+yolov5)

B站 :道传科技上位机 观看教

一、pycharm的安装

pycharm windows版本下载地址:Download PyCharm: The Python IDE for data science and web development by JetBrains

下载社区版本(日常学习使用够用了),专业版是收费的哦(功能更强大)

二、Python的安装

1-1 首先打开www.python.org (这个是python官网)下载配置环境。点击上方 downloads

1-2 根据系统选择python环境下载

三、PYQT5的安装

在终端里面输入

  1. pip install PyQt5

  2. pip install PyQt5-tools

如果安装完成之后没有 designer.exe

需要在终端输入 pip install PyQt5Designer

"名称": 这里可以自定义的

"程序":选择的是 "pyuic5.exe" 的安装目录

"实参":FileDir\FileName

"工作目录": 根据实际情况配置,这里我直接使用的是 宏 FileDir

四、CUDA的安装

  1. 查看显卡型号与驱动版本(Driver Version)

①打开终端(win+R输入cmd),输入

nvidia-smi

2、

大家要了解的是:

1)驱动版本决定了CUDA版本,驱动版本越高,则可安装的CUDA版本越高

2)CUDA向下兼容,我这里是输出的CUDA版本为12.1,并不是意味着我的电脑只能安装12.1版本的CUDA,而是可以安装12.1及以下任何版本的CUDA

2. 下载CUDA

①Org Download WEB:

CUDA Toolkit Archive | NVIDIA Developer

我选择的12.1.0

五、torch 和torchvison 的安装

具体的安装步骤如下:

1) pytorch 官网下载:https://download.pytorch.org/whl/torch_stable.html

打开上面链接,界面如下,以下包含torch、torchvison、torchaudio 等相关包的各个版本,其中每个数字代表的含义入下图:

找到自己想要下载的版本,点击即可下载。

注意:cuda的版本、python的版本、以及windows/linux 这些必须对应好,否则即便安装上,也不能调用cuda。

我们需要下载 torch 和 torchvision 两个关键包,torchaudio是通音频相关的包,不搞音频的话,可以不用安装。

2)下载完成后,得到的是 whl文件, 然后打开文件所在文件夹,在路径下输入cmd,打开 命令提示符, 输入 pip install xxx.whl 进行安装,先安装 torch包、再安装 torchvision包。

六、yolov5的下载

1、.github上下载yolo包

https://github.com/ultralytics/yolov5

2、顺利进入环境后,在终端输入以下代码安装依赖:

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

#这里使用清华源镜像下载,速度更快

然后输入

复制代码
python train.py

等待一会,这里根据电脑性能,运行速度有快慢之分。

如果出现以下类似代码,则恭喜你,代码已经跑通了

2、安装labelimg,进行打标签

在cmd终端中,运行下面代码

复制代码
pip install labelimg

等待下载完成,输入labelimg,即可 运行

相关推荐
DAWN_T1719 分钟前
Python编译器(Pycharm Jupyter)
python·机器学习·jupyter·pycharm
Draina24 分钟前
在pycharm中运行sagemath脚本的配置过程
ide·python·安全·pycharm·密码学
九章云极AladdinEdu25 分钟前
冷冻电镜重构的GPU加速破局:从Relion到CryoSPARC的并行重构算法
人工智能·pytorch·深度学习·机器学习·自然语言处理·架构·gpu算力
clz13145211 小时前
二,神经网络
人工智能·深度学习·神经网络
灵智工坊LingzhiAI2 小时前
基于YOLO的足球检测Web应用:从训练到部署的完整实战
yolo
美狐美颜SDK开放平台2 小时前
未来已来:美颜SDK如何通过深度学习实现个性化美颜形象?
人工智能·深度学习·美颜sdk·直播美颜sdk·视频美颜sdk·直播美颜工具·美颜api
禺垣2 小时前
深度学习模型在C++平台的部署
深度学习
CoovallyAIHub3 小时前
基于YOLOv11的CF-YOLO,如何突破无人机小目标检测?
深度学习·算法·计算机视觉
爱吃土豆的马铃薯ㅤㅤㅤㅤㅤㅤㅤㅤㅤ3 小时前
${project.basedir}延申出来的Maven内置的一些常用属性
java·pycharm·maven
飞天小女警momo4 小时前
YOLO 模型 ONNX 导出与跨平台部署流程
yolo