python - ModuleNotFoundError : No module named 'pegasus'

Question

我想尝试 PEGASUS 来总结文章。 https://github.com/google-research/pegasus
原始 repo 的 README 建议使用 Google Cloud Compute Engine，但我使用 Colaboratory notebook。我是日本的一名英语老师，我希望我的学生能轻松尝试这个软件。他们可以体验机器学习和英语段落摘要。
我遵循了这个指示。 https://github.com/google-research/pegasus/tree/f76b63c2886748f7f5c6c9fb547456d8c6002562#setup
这是我的 colab 笔记本。
https://colab.research.google.com/drive/1p95tZcjhfuCLYh23X3S_gZqRoWVhpIlE?usp=sharing
这是我在笔记本中的代码。

%tensorflow_version 1.x

!git clone https://github.com/google-research/pegasus

!export PYTHONPATH=/content/pegasus

%pip install -r /content/pegasus/requirements.txt

!mkdir /content/pegasus/ckpt

!gsutil cp -r gs://pegasus_ckpt/ /content/pegasus/ckpt/

!python /content/pegasus/pegasus/bin/train.py --params=aeslc_transformer \
--param_overrides=vocab_filename=ckpt/pegasus_ckpt/c4.unigram.newline.10pct.96000.model \
--train_init_checkpoint=ckpt/pegasus_ckpt/model.ckpt-1500000 \
--model_dir=ckpt/pegasus_ckpt/aeslc

然后，我收到此错误消息。

Traceback (most recent call last):
  File "/content/pegasus/pegasus/bin/train.py", line 17, in <module>
    from pegasus.data import infeed
ModuleNotFoundError: No module named 'pegasus'

此错误消息表示 python 无法导入“pegasus”模块，但我使用 !export PYTHONPATH=/content/pegasus 这个命令创建了 python 路径。
请给我一些建议好吗？

Answer 1

你不能使用这个

!export PYTHONPATH=/content/pegasus

改用这个

import os
os.environ['PYTHONPATH'] += ':/content/pegasus'

这里是 an example notebook可以正常运行。
我不使用 /content/pegasus/pegasus .我用 /content/pegasus直接，通过使用 !npx degit .. 安装它