作者：RayChiu_Labloy
版权声明：著作权归作者所有，商业转载请联系作者获得授权，非商业转载请注明出处

目录

本文初衷 

环境准备 

下载pytorch版本的yolov5

pycharm打开yolov5并搭建portch环境

创建conda虚拟环境

安装pytorch依赖包

我采用离线安装方式 

 也可以在线根据电脑配置生成命令安装(这里不采用此方式)

在新建的conda虚拟环境安装whl 

测试torch是否安装成功

 安装yolo requirements.txt依赖

 完成依赖安装

 验证yolov5

搭建win10的pytorch环境并不难，主要是平时在做人工智能项目的时候可能需经常在tensorflow pytorch caffe kaldi 之间来回切换，这个环境搭建都是照官网或者其他博文照着操作即可，我们这里就留一份笔记省的去翻找其他教程了。

环境准备 

我的硬件是 win10  NVIDIA3090 anaconda2019 ,并且安装了cuda11.1 以及cudnn,我准备安装GPU版本的troch

下载pytorch版本的yolov5

github地址 ：   onNX > CoreML > TFLite">GitHub - ultralytics/yolov5: YOLOv5  in PyTorch > onNX > CoreML > TFLite

pycharm打开yolov5并搭建portch环境

创建conda虚拟环境

参考：https://blog.csdn.net/RayChiu757374816/article/details/121313937 

安装pytorch依赖包

我采用离线安装方式 

whl下载地址 https://download.pytorch.org/whl/torch_stable.html

torch和cuda对应关系

 也可以在线根据电脑配置生成命令安装(这里不采用此方式)

地址：Start Locally | PyTorch

在新建的conda虚拟环境安装whl 

pip install E:/torch-1.8.0+cu101-cp37-cp37m-win_amd64.whl

测试torch是否安装成功

import torch
print(torch.cuda.is_available)

 安装yolo requirements.txt依赖

pip install -r requirements.txt -i  https://pypi.tuna.tsinghua.edu.cn/simple

发现报 script file 'D:ProgramDataAnaconda3envspytorch_py37scriptspip-script.py' is not present. 错误，就是没有pip命令，那么conda强制安装pip命令

conda install -f pip

然后再次 pip install -r requirements.txt -i  https://pypi.tuna.tsinghua.edu.cn/simple安装依赖，just waiting some minutes！~

过程中我发现它又给我安装1.10版本的torch去了...(手动狗头)

 完成依赖安装

 验证yolov5

验证自带的数据

python  detect.py --source ./data/images/bus.jpg

会自动下载预训练模型

 效果

 打完收工~~~

 

 

【如果对您有帮助，交个朋友给个一键三连吧，您的肯定是我博客高质量维护的动力！！！】