yolov5 tensort cuda opencv windows10 环境搭建

首先我们需要安装的软件包括vs2017 \cuda\cudnn\opencv\opencv-contrib\anconda\python\cmake\VTK

1、安装显卡驱动 最新即可;

2、先安装cuda\cudnn cuda 10.0版本

3、安装vs2017

4、安装anconda

5、安装python和pytorch环境;

6、安装cmake 最新版本即可;

7、安装VTK源码; 最新版本即可;使用cmake\VS2017编译安装;

8、安装opencv\opencv-contrib ;这两个包都需要下载源码包,并使用cmake\VS2017编译安装; opencv8.4.2版本

TBB 和 cuda 选上;

 

 

 

 

 

9、使用python得到权重文件

10、编译tensort 下的yolov5;并运行转换权重文件;

* 这里需要注意,凡是使用cmake 的一定要选择x64(如果你的是64位系统)编译;

下面这个是我 的环境! 

cmake yolov5得到的工程文件!执行的时候少文件,就需要重新vs编译! 

你可能感兴趣的:(opencv,计算机视觉,python)