AutoDL+SSH在vscode中远程使用GPU训练深度学习模型

注册AutoDL账号

AutoDL官网:AutoDL

注册登录之后,如果你是学生,一定要进行学生认证,可以省钱。 

 认证之后,打开算力市场,

进行GPU选择 

根据自己需要的环境选择版本 ,选好之后创建并开机

这里注意,可以先不开机,开机就会计时收费,先安装插件Remote-ssh

vscode插件

安装之后点击左侧栏图标, 

打开文件后,我们把GPU的SSH指令复制过来。

复制过来后的指令是:ssh -p 31547 root@connect.cqa1.seetacloud.com

然后再点击DL右边的箭头号开始连接

 

 

启动连接,进行初始化 

第一次连接远程的VScode没有下载python,记得在拓展中下载 

使用conda init启动conda,conda list查看torch版本 

 

显示没有scipy,直接pip install scipy即可 ,警告不用管

测试

现在我们来运行一下AlexNet作为测试,epoch为500轮。下图为本地3060跑出的结果。

测试集为102flower数据集,优化器为SGD,学习率为0.01,batch_size = 32。

这是本地3060跑的结果

 远程GPU跑的结果如下图所示,由于时间关系我没有运行完程序,但是可以看出训练时间快了一倍多。

在使用AutoDL的过程中遇到任何问题,可以查看官方帮助文档。可以解决绝大部分问题。 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

HeShen.

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

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

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

打赏作者

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

抵扣说明:

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

余额充值