近期需要经常ssh到远程GPU集群跑代码,每次ssh都要输入密码甚是麻烦,在此记录一下ssh免密码登陆的流程,以及配置环境变量达到进一步偷懒的目的。
我的本地设备是Macbook Pro,需要链接到的GPU集群是Linux系统。首先在本地命令行通过如下命令在本地设备生成秘钥对。
$ ssh-keygen -t rsa
然后console会让你输入存放秘钥的位置,不用输入任何东西直接回车即可放在默认位置。如果已经有了秘钥则会如下面所示,询问是否覆盖。如果没有生成过秘钥,则会直接完成生成。
Generating public/private rsa key pair.
Enter file in which to save the key (/Users/<YOUR_USERNAME>/.ssh/id_rsa):
/Users/<YOUR_USERNAME>/.ssh/id_rsa already exists.
Overwrite (y/n)?
完成生成以后,需要将公钥复制到目标设备,在我这里是将其从本地复制到目标GPU集群的设备上。将如下命令中<username>
替换成目标设备的用户名,<server_IP>
替换成目标ip地址。在本地命令行输入如下命令:
$ ssh-copy-id <username>@<server_IP>
复制的时候需要输入你在目标设备上的账户对应的密码,如果复制成功则会显示:
Number of key(s) added: 1
此时,从本地设备ssh
到目标设备时则无需输入密码。
然而,即便如此我还是觉得十分麻烦,因为我的<username>@<server_IP>
很长,我不愿每次都输入一遍。因此通过配置本地环境变量,可以进一步偷懒。首先在本地命令行输入如下命令,编辑环境变量文件~/.bash_profile
:
$ vim ~/.bash_profile
在里面插入一行内容:
alias <YOUR_COMMAND>="ssh -X <username>@<server_IP>"
# ‘-X' enables the use of X11 forwarding
<YOUR_COMMAND>
可以替换成任意你习惯的命令,用于触发这个ssh命令。保存,然后在命令行通过如下命令更新环境变量:
$ source ~/.bash_profile
此时直接在本地命令行输入<YOUR_COMMAND>
即可直接连接到目标设备。