• 教程 >
  • 在 Google Colab 中运行教程
快捷方式

在 Google Colab 中运行教程

在 Google Colab 中运行教程时,可能需要满足其他要求和依赖项才能使教程正常工作。本节包含有关如何配置各种设置以在 Google Colab 中成功运行 PyTorch 教程的说明。

Google Colab 中的 PyTorch 版本

当您运行需要新发布的 PyTorch 版本的教程时,该版本可能尚未在 Google Colab 中提供。要检查您是否已安装所需的 torch 和兼容的领域库,请运行 !pip list

如果安装的 PyTorch 版本低于要求,请卸载并重新安装,方法是运行以下命令

!pip3 uninstall --yes torch torchaudio torchvision torchtext torchdata
!pip3 install torch torchaudio torchvision torchtext torchdata

在 Colab 中使用来自 Google Drive 的教程数据

我们为教程添加了一项新功能,允许用户在 Google Colab 中打开与教程关联的笔记本。您可能需要将数据复制到您的 Google Drive 帐户才能使更复杂的教程正常工作。

在本例中,我们将演示如何在 Colab 中更改笔记本以与 Chatbot 教程一起使用。为此,您首先需要登录 Google Drive。(有关如何在 Colab 中访问数据的完整说明,您可以查看他们的示例笔记本此处。)

要开始,请在浏览器中打开Chatbot 教程

在页面顶部,单击**在 Google Colab 中运行**。

该文件将在 Colab 中打开。

如果您选择**运行时**,然后选择**运行全部**,您将收到错误,因为找不到该文件。

要解决此问题,我们将把所需文件复制到我们的 Google Drive 帐户中。

  1. 登录 Google Drive。

  2. 在 Google Drive 中,创建一个名为 data 的文件夹,并在其中创建一个名为 cornell 的子文件夹。

  3. 访问 Cornell 电影对话语料库并下载 movie-corpus ZIP 文件。

  4. 在本地机器上解压缩文件。

  5. 将文件 utterances.jsonl 复制到您在 Google Drive 中创建的 data/cornell 文件夹中。

现在我们需要编辑 Colab 中的文件,使其指向 Google Drive 上的文件。

在 Colab 中,将以下内容添加到代码部分开头,位于以 corpus\_name 开头的行的上方。

from google.colab import drive
drive.mount('/content/gdrive')

更改接下来的两行。

  1. corpus\_name 的值更改为 "cornell"

  2. 将以 corpus 开头的行更改为以下内容:

corpus = os.path.join("/content/gdrive/My Drive/data", corpus_name)

现在我们指向了上传到 Drive 的文件。

现在,当您单击代码部分的“运行单元格”按钮时,系统会提示您授权 Google Drive,并且您将获得一个授权码。将代码粘贴到 Colab 中的提示符中,您应该就设置好了。

从“运行时”/“全部运行”菜单命令重新运行笔记本,您将看到它正在处理。(请注意,本教程运行时间很长。)

希望此示例能为您提供一个良好的起点,以便在 Colab 中运行一些更复杂的教程。随着我们在 PyTorch 教程网站上使用 Colab 的发展,我们将寻求方法让用户更轻松地使用它。

启用 CUDA

某些教程需要支持 CUDA 的设备(NVIDIA GPU),这涉及在执行教程之前更改运行时类型。要在 Google Colab 中更改运行时,请在顶部下拉菜单中选择“运行时”,然后选择“更改运行时类型”。在“硬件加速器”下,选择 T4 GPU,然后单击 保存

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源