SOLO模型环境搭建训练流程及问题

一、背景

SOLO是一种图像实例分割模型,我们主要下载的是github这个网站SOLO
因为该模型是基于mmdetection跟mmcv来的,所以要多注意学习掌握这两个东东

二、流程

1.根据install.md安装环境

Requirements

此次安装版本

Python 3.8
pytorch       1.10.2       py3.8_cuda11.3_cudnn8.2.0_0    pytorch

先找到Installation部分,打开INSTALL.md开始安装。首先一定要注意Requirements中的各种版本问题,这里面处处是坑。此次SOLO的训练环境是在从NGC下载的pytorch21.07镜像所起的容器中建立的。

a.创建conda虚拟环境并激活
conda create -n solo python=3.8 -y  #不是3.7,是3.8,当然以后训练的时候不知道是几了
conda activate solo

这里注意,python版本得是3.8了,要求中的3.7跟其他库的对应版本起冲突

b.按照官方说明安装 PyTorch和torchvision

如果直接按照它给的来,应该是直接装上了最新版本,也是可以的
可以根据official instructions挑选所需要的命令

cond
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值