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。
远程GPU跑的结果如下图所示，由于时间关系我没有运行完程序，但是可以看出训练时间快了一倍多。
在使用AutoDL的过程中遇到任何问题，可以查看官方帮助文档。可以解决绝大部分问题。
腾讯云面向开发者汇聚海量精品云计算使用和开发经验，营造开放的云计算技术生态圈。
更多推荐
7
0- 0
已为社区贡献1条内容
所有评论(0)