讲一下第一次使用服务器训练yolov3的全过程

本文详细记录了在Ubuntu服务器上首次使用Anaconda建立虚拟环境,安装PyTorch(CUDA 10.0,驱动410.48),并配置YOLOv3训练目标检测任务的过程。遇到的主要问题是PyTorch版本与CUDA及驱动不匹配,以及安装指定版本PyTorch的困扰,最终通过选择适合服务器配置的PyTorch版本解决问题,成功开始训练。
摘要由CSDN通过智能技术生成

截至目前还没有过一次完整的炼丹经历,这次借着课程大作业的需要 尝试训练一下yolov3网络做一个基于自己数据集的目标检测任务 接下来讲一下从环境配置到完成配置的全过程

1:建立虚拟环境:网上建立虚拟环境的博客很多,但还是觉得使用anaconda建立虚拟环境更加可靠:在实验室的服务器建立账户之后,在自己的账户下先安装anaconda3,(下载好脚本文件后直接在相关文件夹下bash Anaconda3-2020.02-Linux-x86_64.sh)这个教程太多了 在这里不再赘述,安装完成后先配置一下国内的镜像源,配置可以直接抄这里:清华镜像源网址提供的配置描述 配置好国内的镜像源后在建立虚拟环境,下载安装包的时候会非常快捷,之后就可以建立虚拟环境了,可以参考此博客:讲了许多实用的conda命令

2:安装pytorch:这里我耗了将近两天时间,我直接讲成功的流程—:首先我使用的开源项目来自于:github 里面提供了项目所需要的pytorch版本为>=1.4.0Train Custom Data , 我先查看了我的CUDA版本:使用nvcc -v命令,得知CUDA版本为10.0.130。再

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值