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

或者:


网友评论