使用的项目地址:https://github.com/Tianxiaomo/pytorch-YOLOv4
一、配置环境
1.因为不同的项目所需要的环境也不同,每次都更新太费神,所以直接在anaconda下建一个虚拟环境,给环境起名叫 yolov4:
conda create -n yolov4 python=3.7
因为我的 anaconda 自带的 python 就是3.7版本,所以这里安装3.7。
2.激活虚拟环境
上面的命令执行后会出现如下提示:
故我们输入conda activate yolov4
来激活该环境。
若想退出该环境,可以使用:conda deactivate
3.安装pytorch
进入pytorch官网:https://pytorch.org/,按照下图选择好对应的版本:
然后输入如下命令即可:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
出现的小插曲:
因为我的网速太烂,导致最大的三个下载失败,于是乎就折腾了一下午各种换网。
后来开热点,前两个总算是下载好了,然后那个700多兆的大块头一直稳如泰山,我就开始了我的一系列踩坑,记录一下,长点教训:
(1)下载不了报错如下:
我就想,既然这里面下载的慢,链接给我了,我为啥不先下载好呢,于是我复制划线的链接在浏览器打开,下载好了
自信满满的执行了:pip install pytorch-1.7.1-py3.7_cuda102_cudnn7_0.tar.bz2
,然后华丽丽的报错了:
看见这个错误提示一脸懵,根本不知道 pip-req-build-33tnmdbn 是什么,主要是Temp文件夹里面根本没有这个东西,百度也无果,病急乱投医嘛,找到https://blog.csdn.net/weixin_44566432/article/details/109814433这篇文章好像和我的类似,我就学博主的加上了如下的模块:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn
然后,又华丽丽的报错了:
(2)成功的方案:
被校园网坑了一下午,还是老老实实去下载了torch-1.7.1-cp37-cp37m-win_amd64.whl
,然后执行如下命令行:
pip install torch-1.7.1-cp37-cp37m-win_amd64.whl
可算是成功了,然后因为需要其他的环境,都在 requirements.txt 里面,我看有些博主可以直接执行pip3 install -r requirements.txt
这个命令,但是我执行就报错,所以直接输入了pip list
查看
后面运行命令缺啥再补啥吧~