我怎么能ssh到google colaboratory VM？

Question

我如何ssh到谷歌合作VM？有没有比从jupyter笔记本到远程服务器创建ssh隧道更直接的方法呢？

Answer 1

这是另一个解决方案 - 本质上与@tamlyn 提供的相同，但要短得多。有一个colab-ssh连接器，可以自动执行该过程。您所要做的就是在ngrok上注册一个免费帐户。然后，您需要复制您的auth token。现在，在目标 Colab 笔记本中：

# Install colab_ssh
!pip install colab_ssh --upgrade

from colab_ssh import launch_ssh
launch_ssh('YOUR_NGROK_AUTH_TOKEN', 'SOME_PASSWORD')

你应该得到连接参数（注意端口）。现在，您可以使用提供的详细信息连接到 Colab VM，例如：ssh -p 15892 root@0.tcp.ngrok.io，15892您需要更改的随机端口号在哪里。当然，系统会提示您输入先前设置的密码。

这显然受到免费 ngrok 帐户限制的限制，但对于小任务来说应该足够了。

如果你得到了IndexError: list index out of range，别担心。根据我的经验，它仍然可以工作 - 您可以从ngrok active tunnels list获取端口和地址。

Answer 2

尝试将以下内容粘贴到单元中并运行它（受本要点启发）：

import random, string, urllib.request, json, getpass

#Generate root password
password = ''.join(random.choice(string.ascii_letters + string.digits) for i in range(20))

#Download ngrok
! wget -q -c -nc https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip
! unzip -qq -n ngrok-stable-linux-amd64.zip

#Setup sshd
! apt-get install -qq -o=Dpkg::Use-Pty=0 openssh-server pwgen > /dev/null

#Set root password
! echo root:$password | chpasswd
! mkdir -p /var/run/sshd
! echo "PermitRootLogin yes" >> /etc/ssh/sshd_config
! echo "PasswordAuthentication yes" >> /etc/ssh/sshd_config
! echo "LD_LIBRARY_PATH=/usr/lib64-nvidia" >> /root/.bashrc
! echo "export LD_LIBRARY_PATH" >> /root/.bashrc

#Run sshd
get_ipython().system_raw('/usr/sbin/sshd -D &')

#Ask token
print("Copy authtoken from https://dashboard.ngrok.com/auth")
authtoken = getpass.getpass()

#Create tunnel
get_ipython().system_raw('./ngrok authtoken $authtoken && ./ngrok tcp 22 &')

#Get public address and print connect command
with urllib.request.urlopen('http://localhost:4040/api/tunnels') as response:
  data = json.loads(response.read().decode())
  (host, port) = data['tunnels'][0]['public_url'][6:].split(':')
  print(f'SSH command: ssh -p{port} root@{host}')

#Print root password
print(f'Root password: {password}')

ngrok用于创建到计算机的隧道，并为其提供可公开访问的主机名。您将需要注册，复制身份验证令牌，并在出现提示时将其提供给Colab笔记本。

完成后，它将打印出您的连接详细信息，例如：

SSH command: ssh -p12312 root@0.tcp.ngrok.io
Root password: abcdeLMh6vpNViGHQbXi

如果最后一步失败，请尝试再次运行该单元或仅使用以下内容创建一个新单元，然后读取JSON输出以查找主机名和端口。

! curl -s http://localhost:4040/api/tunnels

Answer 3

由于 Google 决定阻止ngrokGoogle Colab，因此您可能需要使用不同类型的 SSH 隧道。你可以试试这个：

!pip3 install linus_colab_ssh

from colab_ssh import setup_ssh, loop_forever

public_key = '<YOUR_PUBLIC_SSH_KEY>'
setup_ssh(public_key)
loop_forever()

更多详细信息可以在这里找到。祝你好运！

Answer 4

您可以使用在笔记本中执行 shell 命令!，例如：

!echo "Echo in shell"

目前不支持通过 SSH 直接连接到后端。