pytorch-yolov4训练自己的数据集之环境配置

使用的项目地址: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查看
在这里插入图片描述
后面运行命令缺啥再补啥吧~

  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 9
    评论
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值