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

相关推荐
m4Rk_3 小时前
【论文阅读】Agent 记忆机制(22):FluxMem——根据对话结构动态选择记忆组织方式
论文阅读·人工智能·学习
YHL3 小时前
🤖 Agent 智能体开发实战 · RAG 实战 —— 从零搭建检索增强生成系统
人工智能
ambition2373 小时前
WorkBuddy迁移至D盘完整操作指南
人工智能
科技圈快迅4 小时前
在机场与高铁站:智能服务机器人正在深度对接业务系统,实现从信息孤岛到移动服务窗口的转变
大数据·人工智能·机器人
小企鹅么么4 小时前
【AI应用开发工程师】第七章:Agent 工程与实践
人工智能
CIO_Alliance4 小时前
AI+iPaaS解决方案:跨系统业务流程自动化助力企业AI化转型
运维·人工智能·自动化
综合资讯4 小时前
2026年打开设备ETF的产品清单:八只同类,两只能买
人工智能·科技·芯片·半导体·科创
声讯电子4 小时前
A-59 双麦双通道语音处理模块:重新定义全双工通话的极限性能
人工智能·语音识别·ai降噪·usb接口·回音消除
菜鸟‍4 小时前
【论文学习】Frontiers in Physics 2024 || Cap2Seg:基于文本描述生成的 COVID-19 医学图像分割性能提升方法
人工智能·学习
陕西企来客4 小时前
生成式引擎优化(GEO)行业白皮书:从流量分配到认知占有的战略跃迁
人工智能·算法·机器学习·技术好geo优化