Visual stdio2022 opencv cude pytroch与yolov8/可视化工具的环境搭建,不搞VIP,我也要当雷锋

顺序很重要,别瞎搞

vs studio2022

编程小白必看!Visual Studio 2022详细安装使用教程(C/C++编译器)-CSDN博客

pycharm

2024最新PyCharm下载安装配置教程,软件安装看这篇就够了!_pycharm安装-CSDN博客

下载解压软件,这里推荐bandi 推荐三款最好用的压缩/解压软件_好解压-CSDN博客

下载python解释器,这里有两种,一种官方下载Welcome to Python.org但是太慢了,这里作者选择国内的python下载Index of python-local之后安装,要注意path,我的版本是3.10.10,因为后面要和pycharm cude等适配这个版本测试过可以跑起来,按照好后cmd测试一下


opencv部分

接下来下载opencv480,同样官网下载太慢,作者选择并找到了雷锋OpenCV下载/OpenCV国内镜像/opencv_contrib下载_opencv镜像下载-CSDN博客

下载好opencv后进行环境的搭建配置

这三个vs2022中的配置,之后配置系统变量

重启!!

之后可以简单写些读取图像的程序测试一下

#include <opencv2/opencv.hpp>

#include <iostream>

int main(int argc, char** argv) {

// 读取彩色图像作为模板

Mat templateImage = imread("C:/image/1.png");

if (templateImage.empty()) {

cout << "无法读取模板图像" << endl;

return -1;

}

namedWindow("New Image with Differences", WINDOW_AUTOSIZE);

imshow("New Image with Differences", newImage);

waitKey(0);

return 0;

}

之后在cmd pip list 如果没有opencv-python 通过镜像下载pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

opencv-python的依赖已经安装好了在pycharm中进行读取图片测试

要记住的是有可能会报错

需要setting一下正确的python解释器

之后进行测试

复制代码
import cv2 as cv
import numpy as np
image=cv.imread("C:/image/1.png")
cv.imshow("imput",image)
cv.waitKey(0)
cv.destroyAllWindows()

Pytorch部分

Pytorch要基于cuda安装,首先查找本机cuda版本,cmd中nvcc-v,如果出现cuda版本,则根据版本在pytorch中找到对应版本后下载,这里作者从没有搭建 cuda开始。

这篇文章写的很清楚cuda要怎么装CUDA安装教程(超详细)-CSDN博客

安装好后看一下

之后根据我们的CUDA11.6去下载pytorch

最好是安装1.13.1的pytorch兼容性最好,但是作者没有找到这个版本的镜像,所以选了1.13.0,后面使用会报错,需要将Numpy降一下版本。后面会说

国内镜像pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

其实只要电脑支持下载什么版本都行但是要注意兼容问题,不然真的很烦,我这个版本尝试过了没问题,所以第一次下载尽量用我这个。

下载好以后测试一下

之后降一下Numpy的版本,有时候没有办法下载要切换一下powershell,方法是:

在CMD窗口中输入powershell并按回车键,即可切换到PowerShell环境。若要从PowerShell切回CMD,输入cmd并回车。

镜像是pip install numpy<2 -i https://pypi.tuna.tsinghua.edu.cn/simple

后面就可以正常使用了,测试一下

复制代码
import torch as t
a=t.tensor([1,2])
b=t.tensor([1,2])
c=a+b
print(c)

yolov8部分

需要安装的依赖

  • win10 64
  • python 3.8.x
  • opencv-python4.6.x
  • pytorch1.12
  • CUDA11.3
  • tensoRT8.4

官网下载https://github.com/ultralytics/ultralytics

作者选择镜像pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple

下载后测试一下

>>> import ultralytics

>>> ultralytics.checks()

如果和作者一样就是安装成功了,我们进一步测试看能否简单识别

首先先建立一个文件夹,用于存储模型和我们的图片

之后进入我们新建的文件夹来下载模型

  1. cd C:\python\my_yolov8_train_demo
  2. yolo predict model=yolov8n.pt source=zidane.jpg
  3. yolo predict model=yolov8n.pt source=zidane.jpg save=True

下载在my_yolov8_train_demo的模型文件和我提前找好的模板

之后运行save=True,会自动生成run文件里面保存生成的文件

NETRON下载

可视化工具的下载,将yolov8n.pt模型导出为ONNX格式

无需多言

Netron下载2025最新pc版_Netron电脑版官方免费下载_华军软件园

下载 onnx onnxslim onnxruntime

镜像 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple onnx onnxslim onnxruntime

import onnx

print(onnx.version)

1.17.0

import onnxslim

print(onnxslim.version)

0.1.48

import onnxruntime

print(onnxruntime.version)

1.21.0

下载成功后执行:yolo export model=yolov8n.pt format=onnx

生成ONNX格式,之后用Netron打开来看模型

下面这个是模型预测

基于ONNX的检测,输入:yolo predict task=detect model=yolov8n.onnx imgsz=640 source=zidane.jpg save=True

可以看出模型没有丢失

相关推荐
分布式存储与RustFS2 分钟前
RustFS的边缘计算优化方案在5G MEC场景下的实测数据如何?
人工智能·5g·开源·边缘计算·rustfs
2501_9248905210 分钟前
商超场景徘徊识别误报率↓79%!陌讯多模态时序融合算法落地优化
java·大数据·人工智能·深度学习·算法·目标检测·计算机视觉
Jia-Hui Su27 分钟前
GDSFactory环境配置(PyCharm+Git+KLayout)
git·python·pycharm
SalvoGao31 分钟前
空转学习 | cell-level 与 spot-level的区别
人工智能·深度学习·学习
初岘34 分钟前
自动驾驶GOD:3D空间感知革命
人工智能·3d·自动驾驶
什么都想学的阿超1 小时前
【大语言模型 15】因果掩码与注意力掩码实现:深度学习中的信息流控制艺术
人工智能·深度学习·语言模型
码蛊仙尊1 小时前
当我们想用GPU(nlp模型篇)
人工智能·自然语言处理
学习3人组1 小时前
手写数字识别代码
人工智能·python
Codebee2 小时前
Qoder初体验:从下载到运行OneCode可视化设计器的完整实战指南
人工智能
双向332 小时前
高并发AI服务部署方案:vLLM、TGI、FastChat性能压测报告
人工智能