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 小时前
Agent 换了 Pod,身份怎么办?拆解 Substrate 的 Actor Identity
人工智能·云原生
人工智能AI技术10 小时前
从ChatGPT到GPT‑6 Astra:当AI开始解数学难题、自主操作软件
人工智能
财迅通Ai10 小时前
物理AI从产业叙事走向基本面重估 以SENASIC琻捷看端侧感算入口的产业价值
人工智能·senasic琻捷
Geek-Chow10 小时前
MCP 模型上下文协议:九、深入服务器 · 工具、资源与提示
人工智能
Theo_xx10 小时前
声学感知基础:Day1.常见音频类型(Chirp、FMCW、OFDM)的区别、联系和选择
人工智能·无线感知·声学感知
DisonTangor10 小时前
Qwen-Drive-1.0:首个统一 3D 感知、视觉问答与运动规划的自动驾驶视觉语言基础模型
人工智能·3d·自动驾驶
XiaoKe202610 小时前
数海云擎:以内外双向数智化锻造竞争壁垒,跑通销售全链路
大数据·人工智能·科技·软件需求
m0_5873830010 小时前
全民健身解决方案软件开发:从架构设计到落地实践
人工智能·数据挖掘·系统架构·需求分析
Zzj_tju10 小时前
小模型指令微调:数据混合、模板与过拟合的最小复现
人工智能·深度学习·机器学习·语言模型
basketball61611 小时前
AI Infra 推理部署技术总结:2. vLLM 内核——PagedAttention 与调度器原理
android·人工智能·vllm·ai infra