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
创建一个conda虚拟环境:
输入conda create -n pytorch_py37 python=3.7 创建一个新的虚拟环境,python版本为3.7,在这个独立的环境中安装pytorch
2. 安装pytorch 1.6
在pytorch官网,根据情况选择安装的conda install命令
若开始下载速度慢,可以考虑将conda的下载源换成国内的清华或者中科大的镜像源,这里本人选择中科大的
等价于在C:\Users\Clark\.condarc文件中输入上述内容
如上图,输出pytorch版本以及cuda可用后,至此,pytorch安装成功
3. 安装YOLOv5
可以直接下载源码zip,发现8月14更新了yolov5的v3.0版本,所以也可以通过右侧releases下载
在Anaconda Prompt命令行中cd到解压后的yolov5-3.0的源码目录下
输入pip install -r requirements.txt 安装必需的python依赖
4. 测试yolov5的效果
直接在Anacoda Prompt中输入命令 python detect.py --source 0
或者: