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

注册AutoDL账号

AutoDL官网：AutoDL

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

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

进行GPU选择 

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

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

vscode插件

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

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

复制过来后的指令是：ssh -p 31547 [e-mail]

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