pycharm用ssh远程连接gpu服务器跑代码

目录

ssh连接

在Pycharm上配置ssh服务

使用gpu服务器运行python程序

gpu与cpu训练速度对比

---

学校老师提供了gpu服务器给学生使用，这里记录一下用gpu跑代码的过程

ssh连接

在本机安装openssh

注意：我这里是用非对称密钥登录，用密码登陆的可以忽略下面3步

生成密钥对：ssh-keygen

生成的公钥保存在：c:\users\用户名\.ssh\id-ras.pub

私钥自己保存，公钥发给老师。老师在服务器上设置好后，返回用户名和服务器IP给学生。

cmd测试ssh连接：ssh 用户名@服务器IP

在Pycharm上配置ssh服务

注意：需要使用pycharm专业版，社区版没有SSH Interpreter

打开File->Setting->SSH，设置gpu服务器的IP地址，用户名，端口号

由于我这里是用密钥登录，所以选择Private key选项；如果是密码登录的话，选择Password即可

在Setting->python interpreter里面选择添加解释器，添加刚刚配置的SSH，然后点击next

然后添加python解释器，并在服务器上创建一个同步目录，意思是本地项目和远程实例中的哪个目录进行关联

回到选择解释器的界面，选择该解释器。

只需要做一次，以后的项目可以直接选择这个解释器，然后设置本地与服务器的目录映射

在Setting->deployment里面可以测试下ssh的连接状况

显示successfully表示ssh连接成功

可以在mapping里设置同步目录（本地项目和远程实例中的哪个目录进行关联）

如果配置均无误PyCharm会有一小会配置过程。

使用gpu服务器运行python程序

配置好后，可以在terminal终端中下拉找到远程服务器的终端

cd跳转到刚刚设置的同步目录中

ls看一下项目文件是否都自动同步过来了

如果在运行时找不到Python文件，可能是没有自动同步代码，可以选择手动同步

提示缺少一些库

在终端里pip install安装

安装完缺失的库后，运行.py代码

gpu与cpu训练速度对比

可以看到用gpu服务器训练一轮大概花费20秒

重新换成自己的电脑的解释器对比一下训练速度，可以看到用电脑cpu训练一轮大概花费40秒

对比显示，用gpu服务器确实能提升训练速度