1、下载jdk并安装； 2、下载anaconda并安装；就不要自己去下载python运行环境了，安装各种包太麻烦了。直接安装个anaconda就省事了； 3、配置环境变量： 用户变量： a、新建JAVA_HOME，内容C:\Program Files\Java\jdk-14.0.2 b、path里面增加%JAVA_HOME%\bin c、path里面增加C:\Users\用户名\anaconda3 d、path里面增加C:\Users\用户名\anaconda3\Library\bin e、path里面增加C:\Users\用户名\anaconda3\Scripts 系统变量同上； 4、下载并安装eclipse；选择eclipse IDE for java Developers就行； 5、启动Eclipse, 点击Help->Install New Software... 在弹出的对话框中，点Add 按钮；Name中输入:Pydev, Location中输入https://pydev-9-2-0.surge.sh a、这得注意，很多文章都说让用https://www.pydev.org/updates/，这个永远都安装不成功。一定得用https://www.pydev.org/updates/提示的https://pydev-9-2-0.surge.sh才行； b、下一步有两个选项，只选择选PyDev；然后Next； c、选择I accept the terms of the license agreements；然后Finish； d、开始Installing Software，弹框的话选择Install anyway； e、安装完成选择Restart Now； 6、重新启动Eclipse后，安装Pydev的解释器： a、点击 window->perperence， 找到PyDev； b、点击PyDev -> Interprters -> Python Interprter； c、点击右侧的 New ... -> Browse for python/pypy.exe -> 选择python安装目录C:\Users\用户名\anaconda3\python.exe -> OK； d、下一个界面全选然后点OK，最后点击Apply，完成配置； e、Window —> Preferences —> General —> Content Types下，将Python File的Default encoding设置为：UTF-8； f、Window —> Preferences —> General —> Workspace中将Text file encoding设置为：UTF-8； 7、下面安装eclipse中文插件，这不是必须的，看你英文怎么样。我开发java的时候就用英文的，开发python不熟所以用中文： a、Help -> Install New Software； b、直接点击Add， c、Name输入Babel，Location输入https://download.eclipse.org/technology/babel/update-site/R0.19.2/2021-12/，这个可以去找找最新版的； d、勾选Babel Language Packs in Chinese (Simplified)； e、一路Next，选择I accept the terms of the license agreements；然后Finish； f、弹出框选择Install anyway； g、安装完成选择Restart Now； 8、把YOLO5工程的整个目录放到C:\Users\用户名\eclipse-workspace\下面 -> 导入 -> 常规 -> 现有项目到工作空间中； 9、找到train.py和detect.py直接运行，运营肯定会报错，没关系，继续往后做； 10、点击 运行 -> 调试配置: train.py -> (x)=Arguments -> Program arguments： --data ../MaskDataSet/data.yaml --cfg models/yolov5s.yaml --weights '' --batch-size 16 detect.py -> (x)=Arguments -> Program arguments： --source ./inference/images/ --weights best.pt --conf 0.4 11、如果运行时出现下面报错就安装对应包： No module named 'cv2' -> pip install opencv-python No module named 'torch' -> pip install torch No module named 'tensorboard' -> pip install tensorboard No module named 'torchvision' -> pip install torchvision 12、如果执行train.py时候报下面这个错： OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program. That is dangerous, since it can degrade performance or cause incorrect results. The best thing to do is to ensure that only a single OpenMP runtime is linked into the process, e.g. by avoiding static linking of the OpenMP runtime in any library. As an unsafe, unsupported, undocumented workaround you can set the environment variable KMP_DUPLICATE_LIB_OK=TRUE to allow the program to continue to execute, but that may cause crashes or silently produce incorrect results. For more information, please see http://www.intel.com/software/products/support/. 就在train.py前面加上os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
【2022年4月1日补充：不是愚人节的玩笑啊】 这篇文章是专门为一个朋友写的。因为他新买的电脑，不太熟悉环境搭建的过程，所以仓促凭印象记录了一下。 上次写道这里就结束了，结果忘记了GPU的事情，结果朋友的电脑明明有gpu，却用cpu训练了好久，实在太不好意思了，所以今天赶紧补上后面这部分。
如果您的电脑有GPU，就先用下面命令检查一下： import torch torch.__version__ 显示：'1.9.0+cpu'
import torch print(torch.cuda.is_available()) 显示：False
运行后输出的是False，说明cuda有问题。如果输出True就没有问题了。
如果输出的False，就先看看自己是不是独立显卡，显卡的驱动安装好了没有。如果显卡驱动确实没有问题，那大概率就是torch、torchvision、torchaudio这三个版本的问题，如果显卡是NVIDIA，就根据自己版本改一下吧。我的显卡是CUDA 9.1，型号是MX150，一款很小众的显卡，所以要用下面这个配置： pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
其实这么看起来，本文上面 No module named 'torch' -> pip install torch No module named 'tensorboard' -> pip install tensorboard No module named 'torchvision' -> pip install torchvision 三项的安装可以先缓一缓了，要根据自己显卡型号来选择版本。
精彩文章
发表评论