2020-10-01 window10 + cuda10.2 +
2020-10-01 本文已影响0人
ClarkWang1214
0. 安装 cuda10.2 + cudnn8.0.4
这一部分配置详见另一篇 Windows10环境下(cuda10.2+cudnn8.0.4)以及Visual Studio2019安装配置
1. 安装 anaconda3:
由于官网下载较慢,我是去清华的镜像源下载(https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/)速度比较快,推荐!
清华大学开源软件镜像站anaconda3 安装到指定目录后,打开开始菜单中的 Anaconda Prompt
打开开始菜单中的 Anaconda Prompt 查看虚拟环境,只有base环境创建一个conda虚拟环境:
输入conda create -n pytorch_py37 python=3.7 创建一个新的虚拟环境,python版本为3.7,在这个独立的环境中安装pytorch
构建一个新的conda虚拟环境,命名为pytorch_py37 激活虚拟环境2. 安装pytorch 1.6
在pytorch官网,根据情况选择安装的conda install命令
在conda虚拟环境中输入此命令进行下载安装若开始下载速度慢,可以考虑将conda的下载源换成国内的清华或者中科大的镜像源,这里本人选择中科大的
在anaconda prompt中输入conda config命令加载中科大的镜像源等价于在C:\Users\Clark\.condarc文件中输入上述内容
等价于修改.condarc文件中的内容 输入上述conda install命令后,完成pytorch的下载 测试pytorch是否安装成功以及gpu是否可用如上图,输出pytorch版本以及cuda可用后,至此,pytorch安装成功
3. 安装YOLOv5
在github上搜索找到第一个ultralytics的最新pytorch版本的yolov5可以直接下载源码zip,发现8月14更新了yolov5的v3.0版本,所以也可以通过右侧releases下载
左边可以直接下载,我是通过进入右侧releases下载最新的Releases版本 拉到最下角,下载源码以及四个不同大小的预训练pt文件,后面训练测试待用在Anaconda Prompt命令行中cd到解压后的yolov5-3.0的源码目录下
找到requirements.txt文件输入pip install -r requirements.txt 安装必需的python依赖
安装必需的python依赖 依赖环境安装成功4. 测试yolov5的效果
直接在Anacoda Prompt中输入命令 python detect.py --source 0
输入命令 python detect.py --source 0 打开外摄的实时对象检测或者:
用源码目录下inference中的图片进行测试 对应多目标检测输出结果图像