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

注册AutoDL账号

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

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

进行GPU选择

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

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

vscode插件

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

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

复制过来后的指令是:ssh -p 31547 [email protected]

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

相关推荐
来一杯龙舌兰17 分钟前
【Bug经验分享】SourceTree用户设置必须被修复/SSH 主机密钥未缓存(踩坑)
运维·缓存·ssh·sourcetree·主机密钥未缓存
未来之窗软件服务18 分钟前
智慧农业、智慧养殖平台—监控摄像头管理监控设计—仙盟创梦IDE
运维·ide·智慧农业·智慧养殖·仙盟创梦ide
深色風信子23 分钟前
Eclipse 插件开发 6 右键菜单
java·ide·eclipse·右键菜单
绝顶大聪明24 分钟前
[特征工程]机器学习-part2
人工智能·机器学习
火星资讯28 分钟前
上海雏鸟科技再赴越南,助力10518架无人机刷新吉尼斯记录
人工智能·科技·机器人
慕婉030740 分钟前
机器学习实战:6种数据集划分方法详解与代码实现
人工智能·深度学习·机器学习·数据集划分
灯下夜无眠1 小时前
sklearn自定义pipeline的数据处理
人工智能·python·机器学习·pipeline·sklearn
Wnq100721 小时前
数据链共享:从印巴空战到工业控制的跨越性应用
大数据·人工智能·数据链共享
-曾牛1 小时前
Azure OpenAI 聊天功能全解析:Java 开发者指南
java·开发语言·人工智能·spring·flask·azure·大模型应用
范桂飓1 小时前
英语口语错题集
人工智能