在服务器上训练自己的yolo模型(包含Xshell的使用)

it2022-12-29  67

  当我们需要一个比较好的GPU而我们的预算又不足的时候,我们可以租用一个服务器用来临时跑自己的深度学习模型,本文以Xshell远程连接GPU服务器的方式介绍如何将自己的模型上传到服务器并运行   1.安装Xshell。在 家庭学校免费版页面下载Xshell的免费版,在当前页面输入姓名与邮件,选中xshell下载即可。免费版的功能可能有阉割,但是对于日常使用是够了的,下载其他的版本可能会存在试用期,试用期过了后需要购买,下载完成后默认安装即可。   2.远程连接服务器。当你获得服务器的ip地址,端口号,用户名以及xshell密码后,打开安装好的xshell软件点击下图所示的新建建立一个新的连接   在当前界面输入服务器的ip地址以及端口号,点击连接   当出现这个界面的时候,选择一次性接受   在弹出的这个界面输入用户名   在这个界面输入连接的Xshell密码   当看到这个界面的时候就是连接成功了   输入命令nvidia-smi查看英伟达显卡信息以及显卡驱动和cuda信息   3.上传训练文件。点击如图所示的文件传输按钮   如果弹出这个界面,点击取消   这会打开一个新的界面,同样会弹出这个对话框,选择一次性接受   使用如下命令打开文件夹,选择要上传到服务器的文件,注意,选择的不能是文件夹,必须是文件,所以如果想上传文件夹,先将文件夹打包压缩即可。选择以后可以看到文件的实时上传速度以及传输完成的总量,上传完成后关闭这个窗口接口

put -h

  4.开始训练。上传到服务器以后,首先使用如下命令解压(以darknet为例)

unzip darknet.zip

  其后面的训练步骤就与在本地训练的步骤一样了(需要注意的是要将数据集准备好以后一起上传,在服务器上执行脚本)如果发现某个脚本错误,可以使用rm命令将其删除,在本地修改以后再将其上传   5.将训练后的权重拿下来。首先安装lrzsz,这个可以把文件上传到服务器或是从服务器上下载下来,使用步骤如下

apt install lrzsz rz #上传,网速比put -h命令慢,将会打开本地资源管理器,选择上传的文件 sz 文件名 #下载,将会打开本地资源管理器,选择下载的位置

  如果你想将一个文件夹下载下来,先将文件夹压缩,再下载即可,压缩命令为(以压缩darknet为例)

zip -q -r darknet.zip darknet

  

最新回复(0)