从 VScode 使用 SSH 时如何启用 GPU 并停止从 colab 启动

Question

我一直在关注这个线程，通过 VSCode 使用 SSH 和 ngrok Configuring Google Colab Like A Pro 进行远程访问。它工作得很好。但是我遇到了两个问题：

第一季度。如何从 VSCode 设置 GPU？在 Colab 中有一个下拉菜单，您可以在其中选择 GPU 来加速训练过程。但是我们如何通过 SSH 在 VScode 中做到这一点呢？

第二季度。如何防止从 Colab 启动？我在Configuring Google Colab Like a Pro 和How to prevent Google Colab from disconnecting 中读到了一个技巧。但它们并不打算通过 SSH 在 VScode 环境中使用。基本上它必须在 Colab 单元中运行。

谢谢。

Answer 1

我找到了解决您面临的问题的 hack [我也遇到了同样的问题 :)]

步骤：

1. 转到 Google Colab，创建一个新笔记本
2. 在单元格中运行此命令!export -p；您应该会看到 Colab 环境可用的环境变量列表。
3. 在输出 declare -x COLAB_GPU="0" 中找到这一行。
4. 在另一个单元格中，运行以下代码：

    import os
    os.environ['COLAB_GPU'] = "1"

5. 再次运行!export -p，现在您应该会看到declare -x COLAB_GPU="1"。
6. 现在运行!pip install colab_ssh --upgrade
7. 安装colab_ssh后，设置一个secret_password并运行以下代码：

    from colab_ssh import launch_ssh_cloudflared
    launch_ssh_cloudflared(secret_password)

8. 按照输出单元格中显示的客户端计算机配置进行操作。

假设您已通过 ssh-tunnel 成功将本地 VSCode IDE 连接到 Colab 内核，您现在应该能够从 GDrive 打开笔记本并运行需要 GPU 访问/高 RAM 的代码。