一. PyTorch环境配置及安装
1.软件安装
安装Anaconda3参考:https://blog.csdn.net/qq_36015370/article/details/79484455
其他部分参考: https://blog.csdn.net/zhouzhiyao960211/article/details/102873717.
注意:3060显卡选择pytorch的v1.80版本的CUDA 11.1
# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
图为Anaconda3默认安装位置
其他快速安装pytorch方法
参考:http://t.csdn.cn/NhfIt
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
2. Pytorch环境安装
1、执行如下的指令查看有哪些环境
conda env list
结果:
2、创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本),如下,我就是创建了一个名字叫YOLOv5 ,python是3.8版本的环境。
conda create -n YOLOv5 python=3.8
结果:
3. pip清华镜像源配置
参考:http://t.csdn.cn/UfkCp
1、临时使用
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
把some-package替换为所需模块即可(还真有个模块叫some-package)
2、永久配置
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
如果报错,可能是因为pip版本不够(需≥10.0.0),先升级pip:
pip install pip -U
当然,升级pip本身也可以使用清华镜像源。
二. 安装YOLOv5依赖
1、按照 “requirements.txt” 安装依赖
//第一步
cd "requirements.txt"文件路径
#例如:cd C:\Users\LZY\Desktop\yolov5\yolov5
//第二步
conda activate pytorch
//第三步
pip install -r requirements.txt
python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ -r requirements.txt
2、安装结果
安装结果给出了pycharm的python配置环境的路径,与上面的环境配置路径一致
上图中的
WARNING: Ignoring invalid distribution...
问题,参考:https://blog.csdn.net/Scarlett2025/article/details/117995427.
警告原因:之前安装插件失败/中途退出,导致插件安装出现异常导致的
解决方法:找到警告信息中报错的目录,然后删掉 “~开头” 的文件夹
三. pycharm打开YOLOv5
1、打开 “文件—设置—Python解释器” 界面,配置python
2、将"当前文件夹"设置为"源根"
链接: https://blog.csdn.net/qq_30622831/article/details/80978118
这一步是解决:pycharm导入本地py文件时,模块下方出现红色波浪线
四. 简单测试
注意:运行v5报以下错误,参考链接:
https://blog.csdn.net/RooKichenn/article/details/120866650.
如果你用的是v5的话,就去v6里面的models/common.py里面去找到这个SPPF的类,把它拷过来到你这个Tags5的models/common.py里面,这样你的代码就也有这个类了,还要在Tags5的models/common.py里引入一个warnings包就可以了
五、安装labelimg
1、注意:首先将pip升级一下,否则有可能出现一些奇奇怪怪的问题
比如报下面错误
Could not find a version that satisfies the requirement sip<7,>=6.4
2、这里主要讲的是在window系统中的安装,首先打开cmd命令行(快捷键:win+R)。进入cmd命令行控制台。输入如下的命令:
pip install labelimg -i https://pypi.tuna.tsinghua.edu.cn/simple
3、有可能出现的问题
(1)按照上述安装有可能会报错:
解决办法:
参考:链接: https://blog.csdn.net/lsf_007/article/details/87931823.
这里利用阿里云安装
python -m pip install --user labelimg -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
(2)安装labelimg过程中出现ModuleNotFoundError: No module named ‘PyQt4’
参考:https://zhuanlan.zhihu.com/p/377302284
// 卸载PyQt5
pip uninstall PyQt5
// 安装python-qt5以获取PyQt5
pip install --user python-qt5