20230117-yolov5训练环境搭建

1.参考资料

https://blog.csdn.net/qq_43573527/article/details/132963466 long错误解决方案
https://pytorch.org/get-started/previous-versions/ pytorch下载的位置

2.服务器环境

conda环境:python 3.7
cuda:11.3

3.安装过程

  • 先创建conda环境,python3.7版本
conda create -n v5 python=3.7
  • 根据cuda版本,安装pytorch
# CUDA 11.3
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
  • 安装v5的依赖包,在安装之前,先把requeirments.txt里面的torch版本和vision版本固定一下。然后安装。
    在这里插入图片描述
    然后安装依赖包,这样的好处是,防止最新的torch版本覆盖已经装的。
pip install -r requirements.txt

到此,按说就可以训练了,但是遇到了问题。

4.问题与解决

在这里插入图片描述
遇到了上述的问题,问题的原因是:新版本的torch无法自动执行转换操作,而旧版本可以。
需要去loss.py中手动改一下,进行long的转换。找到223行,在gain[2],gain[3]后面加.long().就可以正常训练了。

5.补充

一般来说,numpy包和pillow包,都容易因为版本太高报错,遇到乱七八糟的问题,可以尝试降低这俩包的版本

6.其它技巧

基于当前的环境,查找某个包可以安装的版本,该如何查询,比如numpy。只要正常指定版本安装,不输入版本号就能查询到。
在这里插入图片描述

相关推荐

  1. 一步步教您轻松YOLO训练环境(视频教程)

    2024-01-23 04:44:02       11 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-01-23 04:44:02       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-01-23 04:44:02       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-01-23 04:44:02       18 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-01-23 04:44:02       20 阅读

热门阅读

  1. Vue关于router-link的使用和部分代码

    2024-01-23 04:44:02       30 阅读
  2. MySQL存储过程与触发器

    2024-01-23 04:44:02       30 阅读
  3. C++入门

    C++入门

    2024-01-23 04:44:02      34 阅读
  4. H5网页跳转支付宝进行支付

    2024-01-23 04:44:02       41 阅读
  5. 面向对象进阶--接口2

    2024-01-23 04:44:02       38 阅读
  6. 面试百问之mysql 数据库去重的方式有哪些?

    2024-01-23 04:44:02       36 阅读
  7. MYSQL慢查询优化,提升速度

    2024-01-23 04:44:02       36 阅读
  8. 遍历数组的列

    2024-01-23 04:44:02       36 阅读