📜  在 google colab 中连接到本地运行时 (1)

📅  最后修改于: 2023-12-03 14:50:55.903000             🧑  作者: Mango

在 Google Colab 中连接到本地运行时

Google Colab 是一款免费的 Jupyter Notebook 服务,可以在浏览器中运行 Python 代码,它的优势在于能够使用 Google 的强大计算资源。虽然 Google Colab 已经提供了强大的计算资源,但当我们需要使用本地资源时,我们可以通过 "连接到本地运行时" 功能,将本地计算资源连接到 Google Colab 中。

步骤

以下是如何使用 "连接到本地运行时" 功能的步骤:

  1. 在Google Colab中,选择菜单栏上的“连接”->“连接到本地运行时”菜单选项。

  2. 弹出一个对话框,输入你的公共IP地址和端口号,然后点击"连接"按钮。如果你不知道如何获得公共IP地址,可以使用百度查找。

  3. 如果您的本地运行时有已运行的Python内核,则可以选择使用该Python内核。 (如果您没有Python内核,则需要在本地计算机上安装它)

  4. 在Colab笔记本中,您将会看到一个新的代码单元格,你可以在这里输入代码并进行测试,结果会直接返回到 Colab 中。

  5. 在您执行完代码之后,您可以断开与本地运行时的连接。

以下是示例代码来测试本地运行时是否连接成功:

import tensorflow as tf 
with tf.device('/device:GPU:0'): 
  a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a') 
  b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b') 
  c = tf.matmul(a, b) 
with tf.Session() as sess: 
  print (sess.run(c)) 

当您在 Colab 中运行此代码时,它将使用您本地计算机中的GPU运行代码。

结论

使用 "连接到本地运行时" 功能,可以利用本地计算资源来加速计算,同时还能够保持在Colab环境中的工作流程。这为使用 Colab 提供了进一步的灵活性和可用性。