Shortcuts

在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驱动器中的教程数据

我们为教程添加了一项新功能,允许用户在Google Colab中打开与教程相关的笔记本。您可能需要将数据复制到您的Google驱动器账号中,以使更复杂的教程工作。

在此示例中,我们将演示如何更改Colab中的笔记本以适应聊天机器人教程。为此,您首先需要登录到Google驱动器。(有关如何在Colab中访问数据的完整说明,可以查看他们的示例笔记本`here <https://colab.research.google.com/notebooks/io.ipynb#scrollTo=XDg9OBaYqRMd>`__。)

首先在浏览器中打开`Chatbot Tutorial <https://pytorch.org/tutorials/beginner/chatbot_tutorial.html>`__。

在页面顶部点击**Run in Google Colab**。

文件将在Colab中打开。

如果选择**Runtime**,然后选择**Run All**,您会看到文件无法找到的错误。

为了解决此问题,我们将需要将所需文件复制到我们的Google驱动器账号中。

  1. 登录Google驱动器。

  2. 在Google驱动器中创建一个名为``data``的文件夹,并包含一个名为``cornell``的子文件夹。

  3. 访问Cornell Movie Dialogs Corpus并下载电影语料库ZIP文件。

  4. 在本地计算机上解压文件。

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

接下来,我们需要编辑Colab中的文件以指向Google驱动器上的文件。

在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)

我们现在指向上传到驱动器的文件。

现在当您单击该代码部分的**运行单元格**按钮时,系统会提示您授权Google驱动器,并会为您提供授权代码。将代码粘贴到Colab中的提示框中,设置即可完成。

从**Runtime** / **Run All**菜单命令再次运行笔记本,您会看到它开始处理。(请注意,这个教程运行时间较长。)

希望这个示例能够为您在 Colab 中运行一些更复杂的教程提供一个良好的起点。随着我们在 PyTorch 教程网站上对 Colab 的使用不断发展,我们会考虑为用户提供更简单的使用方式。

启用 CUDA

一些教程需要使用支持 CUDA 的设备(NVIDIA GPU),这需要在运行教程之前更改运行时类型。在 Google Colab 中,可以在顶部下拉菜单中选择 Runtime,然后选择 Change runtime type。在 Hardware accelerator 下选择 T4 GPU,然后点击 Save

文档

访问 PyTorch 的详细开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源