1.首先安装Anaconda 3 完成后记得编辑环境变量
win+x 打开管理员powershell,输入 conda init初始化
2.安装Pycharm
3.安装vs-buildtools
4.安装CUDA11.0+CUDNN8.1.0.77
5.创建虚拟环境python=3.8
6.下载yolov5,requirements中先去掉torch,torchvision,cocoapi
7.pip install -r requirements.txt
8.安装pycocotools
9.安装pytorch1.7.1-cuda11.0
10.pip install torchvision
注意:
要成功调用CUDA需要下载pytorch而非torch
Anaconda3+Pycharm+pytorch+CUDA/CUDNN+python3.8的yolov5安装
最新推荐文章于 2024-06-06 14:10:20 发布