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的过程中遇到任何问题,可以查看官方帮助文档。可以解决绝大部分问题。

相关推荐
天空'之城15 小时前
多智能体通信灾难:消息泛滥如何拖垮整个 Agent 系统
大数据·人工智能
赤龙ERP15 小时前
AI+ERP:Agent 进企业,卡在数据和权限这两道坎
人工智能·erp·企业数字化
智圣新创0115 小时前
地方应用型高校智慧校园效能跃升 智圣新创数据治理全链路落地参考框架
人工智能
老饼讲解-BP神经网络15 小时前
一篇详解-BP神经网络matlab实现例子与方法(newff)
人工智能·神经网络·matlab
没落之王15 小时前
AI在文言文——论文脉复苏对AI的意义
大数据·人工智能·算法
司南-704915 小时前
AI infra 学习笔记(三):预处理·第一步——QKV 投影到底算了多少
人工智能·笔记·学习
龙亘川15 小时前
政法工作系统数字化建设实践解读:以全周期闭环驱动政法工作现代化
大数据·人工智能·智慧城市·开源软件·数据可视化
小宋102115 小时前
RAG 知识库也会被投毒:恶意文档、间接 Prompt Injection 与入库审核
人工智能·prompt
AI创界者15 小时前
Python 进阶:重构经典设计模式(六)—— 享元模式与单例模式在 Python 3.10+ 中的内存优化与线程安全演化
人工智能·aigc
honsor15 小时前
PoE温湿度传感器:一根网线供电+通信,即插即用,机房/配电室/仓库温湿度监测首选
运维·服务器·网络·数据库·人工智能·安全