在使用服务器时,配置torch环境(这里是以tengxunyun为例子配置的环境)

这里配置的环境是使用的bubbliiiing的代码为例子进行配置的。配置环境需要对应自己电脑的显卡以及CUDA版本,最重要是torch的安装,一定要去官网去找对应的版本。

1.conda create -n pytorch python=3.6→y→→conda activate pytorch→进入环境安装库


pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

pip install scipy==1.2.1
pip install numpy==1.17.0
pip install matplotlib==3.1.2
pip install opencv_python==4.1.2.30
pip install torch==1.2.0
pip install torchvision==0.4.0
pip install tqdm==4.60.0
pip install Pillow==8.2.0
pip install h5py==2.10.0
pip install future
pip install tensorboard

2.修改检测器中的参数

UnFreeze_Epoch      = 20000
Unfreeze_batch_size = 64
save_period         = 500

3.进入WinSCP

主机名→1.14.246.66
用户名→ubuntu
密码:*************

 登录→是

4.在WinSCP新建→目录→目录名为code(此处误把code打成了conda ,反正就是一个目录的名字)→进入code目录里面

5.上传自己的训练文件,如E:\Fruit_recognition\Grape\SSD,右键SSD→上传→后台上传

上传前的图片

上传后的图片

 将修改后的train文件重新上传。

6.进入终端

cd→cd conda(如果一不下心退到了本地,可以用ssh ubuntu@1.14.246.66进入里面)→ cd SSD→python train.py

7.备注

如果出现库缺失的现象,可以按照提示安装库就可以了,因为之前的库安装的位置可能不对

如果出现训练文件不存在的问题,如下图

重新运行一下python voc_annotation.py

然后在运行python train.py文件即可 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

神笔馬良

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值